pytorch masked_fill 报错的解决. 如下所示: import torch.nn.functional as F import numpy as np a = torch.Tensor([1,2,3,4]) a = a.masked_fill(mask = torch.
確定! 回上一頁