为什么torch.dot函数不能对tensor的矩阵进行点乘
时间: 2023-05-24 12:06:22 浏览: 128
pytorch中torch.max和Tensor.view函数用法详解
5星 · 资源好评率100%
因为torch.dot函数是用于两个一维向量的点乘,它要求输入的两个向量的维度必须是一样的。而对于矩阵的点乘,可以使用torch.mm或torch.matmul函数。这些函数可以对两个二维 tensor 进行点乘,其中 torch.mm 可以适用于两个矩阵的维度都是 2 的情况,而 torch.matmul 可以适用于两个矩阵的维度可以是 2 或者高维。
阅读全文