torch.Tensor.index_fill_¶ Tensor.index_fill_(dim, index, value) → Tensor¶ 通过按 index 中给定的顺序选择索引,使用值 value 填充 self 张量的元素。 参数 dim (int) – 索引的维度 index (LongTensor) – 要填充的 self 张量的索引 value (float) – 要填充的值 示例:>>> x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]], dtype=torch.float) >>> index = torch.tensor([0, 2]) >>> x.index_fill_(1, index, -1) tensor([[-1., 2., -1.], [-1., 5., -1.], [-1., 8., -1.]])