如何转换布尔手电筒Tensor：
>>> t=torch.tensor([0, 1, 0, 1], dtype=torch.bool) tensor([False, True, False, True])
一个包含整数a表示真b表示假的Tensor预期结果：
a
b
tensor([b, a, b, a])
4uqofj5v1#
使用蒙版步进：
mm = torch.tensor([False, True, False, True]) oo = torch.ones([4]) oo[mm] = 42 oo tensor([ 1., 42., 1., 42.])
1条答案
按热度按时间4uqofj5v1#
使用蒙版步进：