Function (autodiff)

function

This is where the forwards and backwards passes live.

Classes:

• Contiguous –
• ContiguousBackward –
• Cast –
• Reciprocal –
• Sin –
• Relu –
• Log –
• Exp –
• Sqrt –
• Sign –
• Less –
• Xor –
• Add –
• Mul –
• Where –
• Sum –
• Max –
• Expand –
• Reshape –
• Permute –
• Pad –
• Shrink –
• Flip –

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

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