假设我有两个张量:
a = torch.randn(10, 1000, 1, 4)
b = torch.randn(10, 1000, 6, 4)
第三个索引是向量的索引。
我想要针对向量a
,计算每个向量在b
中的点积。
具体来说,我的意思是这样的:
dots = torch.Tensor(10, 1000, 6, 1)
for b in range(10):
for c in range(1000):
for v in range(6):
dots[b,c,v] = torch.dot(b[b,c,v], a[b,c,0])
我应如何使用torch函数实现这一点?