快捷方式

torch.Tensor.masked_scatter_

Tensor.masked_scatter_(mask, source)

source 中的元素复制到 self 张量中,位置为 mask 为 True 的位置。从 source 中的元素被复制到 self 中,从 source 的位置 0 开始,并依次逐个继续,对于 mask 为 True 的每个出现情况。 mask 的形状必须与基础张量的形状可广播source 应该至少包含与 mask 中 1 的数量一样多的元素。

参数
  • mask (BoolTensor) – 布尔掩码

  • source (Tensor) – 要从中复制的张量

注意

maskself 张量起作用,而不是对给定的 source 张量起作用。

示例

>>> self = torch.tensor([[0, 0, 0, 0, 0], [0, 0, 0, 0, 0]])
>>> mask = torch.tensor([[0, 0, 0, 1, 1], [1, 1, 0, 1, 1]], dtype=torch.bool)
>>> source = torch.tensor([[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]])
>>> self.masked_scatter_(mask, source)
tensor([[0, 0, 0, 0, 1],
        [2, 3, 0, 4, 5]])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源