把矩阵x中大于5的变成5了
import torch x = torch.linspace(1, 27, steps=27).view(9, 3) bbb = torch.where(x > 5, torch.full_like(x, 5), x) print(bbb)
把矩阵x中大于5的变成5了
import torch x = torch.linspace(1, 27, steps=27).view(9, 3) bbb = torch.where(x > 5, torch.full_like(x, 5), x) print(bbb)