Я хочу использовать pdist. Но форма входного тензора (batch, dim_1, dim_2), и pdist поддерживает только 2D-тензор. Есть ли способ применить эту функцию, но не использовать для l oop, как показано ниже?
vectors = []
for tensor_2D in input_tensor:
vectors.append(torch.nn.functional.pdist(tensor_2D))
tensor = torch.stack(vectors)