masked_fill _(mask, value) 掩码操作用value填充tensor中与mask中值为1位置相对应的元素。mask的形状必须与要填充的tensor形状一致。 a = torch.randn(5,6) x = [5,4,3 ...
確定! 回上一頁