torch.matmul
时间: 2023-10-18 21:10:30 浏览: 88
pytorch:torch.mm()和torch.matmul()的使用
`torch.matmul`是PyTorch中的矩阵乘法函数。它可以用于两个张量之间的矩阵乘法,也可以用于高维张量之间的批量矩阵乘法。其语法为:
```python
torch.matmul(input, other, out=None)
```
其中`input`和`other`是两个张量。如果它们是二维张量,则执行标准的矩阵乘法;如果它们的维度大于二维,则执行批量矩阵乘法。如果给定了`out`参数,则将结果存储在该张量中,否则将创建一个新的输出张量来存储结果。
阅读全文