是将mask中为1的 元素所在的索引,在a中相同的的索引处替换为 value importtorcha=torch.tensor([[[5,5,5,5],[6,6,6,6],[7,7,7,7]],[[1,1,1,1],[2,2,2,2],[3,3,3,3]]])print(a)print(a.size())print("###3")mask=torch.ByteTensor([[[1],[1],[0]],[[0],[1],[1]...
safe_target = target.clone() safe_target.masked_fill_(target_mask,0)ifweights.dim() ==0: weights_to_scatter = Variable(ggI.data.new(safe_target.size()).fill_(1))else: weights_maybe_resized = weightswhileweights_maybe_resized.dim() < target.dim(): weights_maybe_resized = weights_mayb...
minus_mask = (1- mask).byte().unsqueeze(2) attended_arcs.masked_fill_(minus_mask, -numpy.inf)# Compute the heads greedily.# shape (batch_size, sequence_length)_, heads = attended_arcs.max(dim=2)# Given the greedily predicted heads, decode their dependency tags.# shape (batch_size, ...