torch 矩阵相乘

import torch
import numpy as np

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

print(
    '\nnumpy: ',np.matmul(data,data),
    '\ntorch: ',torch.mm(tensor,tensor)
)


numpy:  [[ 7 10]
 [15 22]] 
torch:  
  7  10
 15  22
[torch.FloatTensor of size 2x2]
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容