TensorData

class mint.types.TensorData ()

字段：

• dtype ( TensorDtype )
• shape ( Optional[List[int]] ) – 可选。

张量的形状（见 PyTorch tensor.shape）。长度为 N 的一维列表的形状为 (N,)。若未提供通常可推断得到，一般推断为一维张量。 * sparse_crow_indices ( Optional[List[int]] ) – 可选的 CSR 压缩行指针。设置后该张量为稀疏 CSR：

• data 仅包含非零值（已展平）
• sparse_crow_indices 包含行指针（长度 = nrows + 1）
• sparse_col_indices 包含列索引（长度 = nnz）
• shape 为必填，指定稠密形状
• sparse_col_indices ( Optional[List[int]] ) – 可选的 CSR 列索引。必须与 sparse_crow_indices 同时设置。

property data

展平后的张量数据，以数值数组形式给出。

返回： Union[List[int], List[float]]

from_torch_sparse ( tensor )

从稠密二维 torch 张量创建稀疏 CSR TensorData。

自动检测稀疏性，当能节省空间时以 CSR 编码。张量为一维或大部分非零时回退到稠密。

参数：

• tensor ( torch.Tensor )

返回： TensorData

to_numpy ()

将 TensorData 转换为 numpy 数组。

返回： npt.NDArray[Any]

to_torch ()

将 TensorData 转换为 torch 张量。

返回： torch.Tensor

被引用

• mint.types.ForwardBackwardOutput