我有这样Tensor：
arr1 = np.array([[ 1.6194, -0.6058, -0.8012], [ 1.1483, 1.6538, -0.8062]])
arr2 = np.array([[-0.3180, -1.8249, 0.0499], [-0.4184, 0.6495, -0.4911]])
X = torch.Tensor(arr1)
Y = torch.Tensor(arr2)我想torch.dot对2DTensor中的每个1DTensor（2个向量）执行www.example.com
torch.dot(X, Y)我想得到这样的结果Tensor（[dotResult1，dotResult2]）。但是我得到了这样的错误：
RuntimeError: 1D tensors expected, but got 2D and 2D tensors我的主要目的是对矩阵中的每一个向量做一些运算，但我不想在这里使用循环，有人知道怎么做吗？
1条答案
按热度按时间xpcnnkqh1#
假设您要寻找的是Tensor：
[torch.dot(X[0], Y[0]), torch.dot(X[1], Y[1])]然后你可以做：
测试：
输出：