Pytorch 替换tensor中大于某个值的所有元素

Pytorch 替换tensor中大于某个值的所有元素

使用PyTorch，可以使用以下代码将张量中大于某个值的所有元素替换为新的值：

import torch
# 创建一个示例张量
tensor = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 定义要替换的阈值
threshold = 5
# 将大于阈值的元素替换为新的值
tensor[tensor > threshold] = new_value
# 打印结果张量
print(tensor)

在这个例子中，所有大于阈值 threshold 的元素将被替换为新的值 new_value。