火炬张量将负数设置为零

Question

x=torch.Tensor({1,-1,3,-8})

如何转换x，使得x中的所有负值都替换为零，而不使用循环，使得张量必须看起来像

th>x 1 0 3 0

Answer 1

Pytorch在这里负责广播：

animation=

Answer 2

Pytorch支持操作员索引

a = torch.Tensor([1,0,-1]) a[a < 0] = 0 a tensor([1., 0., 0.])

Answer 3

实际上，此操作等效于应用ReLU非线性激活。

做到这一点，你就很好

output = torch.nn.functional.relu(a)

您也可以就地进行以加快计算速度：

torch.nn.functional.relu(a, inplace=True)