pytorch masked_fill. 原創 Answerlzd 2019-09-16 21:35 ... Tensor([1,2,3,4]) a = a.masked_fill(mask = torch.ByteTensor([1,1,0,0]), value=-np.inf) print(a) b ...
確定! 回上一頁