您可以使用torch.clamp(input, min, max)函数来将张量中小于0的元素变成0。
以下是一个示例代码：
import torch # 创建一个包含负数的张量 x = torch.tensor([-1, 2, -3, 4, -5]) # 将小于0的元素变成0 x = torch.clamp(x, min=0) print(x)
输出：
tensor([0, 2, 0, 4, 0])
在这个示例中，我们使用了 torch.clamp() 函数来将张量 x 中小于0的元素变成0，并打印输出结果。
