b = torch.where(a == 1, torch.Tensor([-1]), a)
torch.where用法