build/torch210-cxx11-rocm71-x86_64-linux/__init__.py

from typing import Optional

import torch

from ._ops import ops

from . import layers


def relu(x: torch.Tensor, out: Optional[torch.Tensor] = None) -> torch.Tensor:
    if out is None:
        out = torch.empty_like(x)
    ops.relu(out, x)
    return out

__all__ = ["relu", "layers"]