pytorch.mm
时间: 2023-11-13 21:58:54 浏览: 190
pytorch:torch.mm()和torch.matmul()的使用
pytorch.mm是一个用于计算两个二维矩阵相乘的函数。它的输入是两个二维张量,输出是它们的矩阵乘积。需要注意的是,两个输入张量的维度必须满足矩阵乘法的规则,即第一个张量的列数必须等于第二个张量的行数。如果输入的张量不是二维的,则需要使用其他函数,如torch.matmul()。在深度学习中,pytorch.mm常用于计算全连接层的前向传播。
阅读全文