pytorch masked_fill 报错的解决 · import torch.nn. · import numpy as np · a = torch.Tensor([1,2,3,4]) · a = a.masked_fill(mask = torch. · print(a) · b = F.softmax(a) ...
確定! 回上一頁