pytorch - 是否有任何内置函数可以使用特定计算从 1D 张量生成 2D 张量？

Question

嗨，我是刚开始深度学习的学生。

例如，我有 1-D 张量x = [ 1 , 2]。从这个，我希望制作第 th 元素具有值y的2D 张量，即，。(i,j)(x[i] - x[j])y[0,:] = [0 , 1]y[1,:]=[ -1 , 0]

pytorch 库中是否有这样的内置函数？

谢谢。

标签： pytorch

Answer 1

在这里，您需要正确的张量暗度才能获得可以使用的预期结果torch.unsqueeze

x = torch.tensor([1 , 2])
y = x - x.unsqueeze(1)
y
tensor([[ 0,  1],
        [-1,  0]])