摘要: https://blog.csdn.net/laox1ao/article/details/79159303 发现对于torch.matmul和torch.bmm,都能实现对于batch的矩阵乘法: a = torch.rand((2,3,10))b = torch.rand((2,2,10))## 阅读全文
posted @ 2019-08-09 10:03 交流_QQ_2240410488 阅读(929) 评论(0) 推荐(0) 编辑