#将 mask必须是一个 ByteTensor 而且shape必须和 a一样 并且元素只能是 0或者1 ,是将 mask中为1的 元素所在的索引,在a中相同的的索引处替换为 value  ,mask value必须同为tensor 

a=torch.tensor([1,0,2,3])
    # 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])

Logo

一站式虚拟内容创作平台,激发创意,赋能创作,进入R空间,遇见同道,让优质作品闪耀发光。​

更多推荐