已给予:
- 张量A的形状为
(batch_size, dim)
- 张量B的形状为
(batch_size, N, dim)
我想计算A中的每一行和B中具有形状(N, dim)
的对应行之间的欧几里得距离
预期结果的形状为(batch_size, N)
已给予:
(batch_size, dim)
(batch_size, N, dim)
我想计算A中的每一行和B中具有形状(N, dim)
的对应行之间的欧几里得距离
预期结果的形状为(batch_size, N)
您应该使用torch.cdist
:
>>> torch.cdist(A[:,None],B)