关于pytorch中部分矩阵乘法的总结(torch.mm,torch.mul,torch.matmul)

1、torch.mul 该乘法可简单理解为矩阵各位相乘,一个常见的例子为向量点乘,源码定义为torch.mul(input,other,out=None)。其中other能够为一个数也能够为一个张量,other为数即张量的数乘。 该函数可触发广播机制(broadcast)。html tensor1 = 2*torch.ones(1,4) tensor2 = 3*torch.ones(4,1) pr
相关文章
相关标签/搜索