PyTorch：具有批处理模式的torch.nn.functional.pdist

Question

我想使用pdist。但是输入张量的形状是（batch，dim_1，dim_2），而pdist仅支持2D张量。有没有一种方法可以应用此功能，但不能像下面这样使用for循环？

vectors = []
for tensor_2D in input_tensor:
    vectors.append(torch.nn.functional.pdist(tensor_2D))
tensor = torch.stack(vectors)