1-2 Tensor的属性

最新推荐文章于 2024-10-23 09:12:35 发布

_卷心菜_

最新推荐文章于 2024-10-23 09:12:35 发布

阅读量272

点赞数

分类专栏： pytorch 文章标签：深度学习

本文链接：https://blog.csdn.net/Thumb_/article/details/114279590

版权

pytorch 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

Tensor的属性

Tensor的属性——稀疏的张量

torch.sparse_coo_tensor
coo类型表示了非零元素的坐标形式

indices = torch.tensor([[0,1,1],[2,0,2]])   # 非零元素的坐标（0，2），（1，0），（1，2）
values = torch.tensor([3,4,5],dtype=torch.float32)  # 对应非零元素的值3，4，5
x = torch.sparse_coo_tensor(indices,values,[2,4])  # 一个 2 X 4 的 tensor