... b.shape) c = np.sqrt(np.power((broadcast_a - broadcast_b.transpose((1, 0, 2))), 2)\ .sum(axis=-1)).T. In Pytorch. import torch a = torch.tensor([[0,0], ...
確定! 回上一頁