我有一个3d张量,其中包含一些零值和非零值。我想将所有非零值替换为零,并将零值替换为特定值。我该怎么办?
几乎完全可以使用numpy做到这一点,就像这样:
tensor[tensor!=0] = 0
为了替换零和非零,您可以将它们链接在一起。只需确保使用张量的副本即可,因为它们已被修改:
def custom_replace(tensor, on_zero, on_non_zero):
# we create a copy of the original tensor,
# because of the way we are replacing them.
res = tensor.clone()
res[tensor==0] = on_zero
res[tensor!=0] = on_non_zero
return res
并像这样使用它:
>>>z
(0 ,.,.) =
0 1
1 3
(1 ,.,.) =
0 1
1 0
[torch.LongTensor of size 2x2x2]
>>>out = custom_replace(z, on_zero=5, on_non_zero=0)
>>>out
(0 ,.,.) =
5 0
0 0
(1 ,.,.) =
5 0
0 5
[torch.LongTensor of size 2x2x2]
使用
torch.where(<your_tensor> != 0, <tensor with zeroz>, <tensor with the value>)
示例:
>>> x = torch.randn(3, 2)
>>> y = torch.ones(3, 2)
>>> x
tensor([[-0.4620, 0.3139],
[ 0.3898, -0.7197],
[ 0.0478, -0.1657]])
>>> torch.where(x > 0, x, y)
Tensor([[ 1.0000, 0.3139],
[ 0.3898, 1.0000],
[ 0.0478, 1.0000]])