Skip to content

Tensor¤

Tensor ¤

Tensor(
    data: (
        ConstType
        | bytes
        | list
        | tuple
        | UOp
        | "np.ndarray"
        | Path
        | None
    ),
    device: str | tuple | list | None = None,
    dtype: DTypeLike | None = None,
    requires_grad: bool | None = None,
)

Bases: SimpleMathTrait

A Tensor is a multi-dimensional matrix containing elements of a single data type.