您的位置:首页 > 产品设计

torch.spmm矩阵乘法

2022-03-15 19:45 1906 查看

Example:

import torch
indices = torch.tensor([[0,1],
[0,1]])
values = torch.tensor([2,3])
shape = torch.Size((2,2))
s = torch.sparse.FloatTensor(indices,values,shape)
print(s)

d = torch.tensor([[1,2],
[3,4]])

print(d)
print(torch.spmm(s,d))
"""
tensor(indices=tensor([[0, 1],
[0, 1]]),
values=tensor([2, 3]),
size=(2, 2), nnz=2, layout=torch.sparse_coo)
tensor([[1, 2],
[3, 4]])
tensor([[ 2,  4],
[ 9, 12]])
"""

 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐