跳至内容

张量¤

张量 ¤

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,
)

基础类: SimpleMathTrait

一个Tensor是一个包含单一数据类型元素的多维矩阵。