pytorch masked_fill :为什么我不能屏蔽所有零? ... mask = mask.unsqueeze(1) scores = scores.masked_fill(mask==0, -np.inf). 也许面具错了?
確定! 回上一頁