Function (autodiff)

function

This is where the forwards and backwards passes live.

Contiguous

Contiguous(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

ContiguousBackward

ContiguousBackward(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Cast

Cast(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Reciprocal

Reciprocal(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Sin

Sin(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Relu

Relu(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Log

Log(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Exp

Exp(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Sqrt

Sqrt(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Sigmoid

Sigmoid(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Sign

Sign(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Less

Less(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Xor

Xor(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Add

Add(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Mul

Mul(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Where

Where(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Sum

Sum(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Max

Max(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Expand

Expand(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Reshape

Reshape(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Permute

Permute(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Pad

Pad(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Shrink

Shrink(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function

Flip

Flip(
    device: Union[str, Tuple[str, ...]],
    *tensors: Tensor,
    metadata: Optional[Metadata] = None
)

Bases: Function