Tensor

Tensor

Tensor(
    data: Union[
        None,
        ConstType,
        List,
        Tuple,
        LazyBuffer,
        "np.ndarray",
        bytes,
        MultiLazyBuffer,
        UOp,
        Path,
    ],
    device: Optional[Union[str, tuple, list]] = None,
    dtype: Optional[DTypeLike] = None,
    requires_grad: Optional[bool] = None,
)

Bases: SimpleMathTrait

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