a.masked_fill(mask = torch.ByteTensor([1,1,0,0]), value=torch.tensor(-1e9)). # tensor([-1.0000e+09, -1.0000e+09, 2.0000e+00, 3.0000e+00]) ...
確定! 回上一頁