Torch 张量将负数设置为零

Question

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

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

th>x 1 0 3 0

Answer 1

Pytorch 支持运算符索引

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

张量([1., 0., 0.])

Answer 2

实际上，这个操作相当于应用ReLU非线性激活。

只要这样做就可以了

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

您还可以就地执行此操作以加快计算速度：

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

Answer 3

Pytorch 负责这里的广播：

x = torch.max(x,torch.tensor([0.]))