在PyTorch中,对于一个张量,如何快速为多个元素赋值相同的值
我们以“a = torch.arange(12).reshape((3, -1))”为例,a里面现在是:
如果我们想让a的右下角的2行3列的元素都为10的话,可以如何快速实现呢?
我们可以用到索引和切片技术,执行如下的指令即可达到目标:
a[1:3, 1:] = 10
执行后的结果为:
这样,我们就达到了预期。
我们以“a = torch.arange(12).reshape((3, -1))”为例,a里面现在是:
如果我们想让a的右下角的2行3列的元素都为10的话,可以如何快速实现呢?
我们可以用到索引和切片技术,执行如下的指令即可达到目标:
a[1:3, 1:] = 10
执行后的结果为:
这样,我们就达到了预期。