web-dev-qa-db-ja.com

すべてのゼロ以外の値をゼロで置き換え、すべてのゼロ値を特定の値で置き換えます

ゼロと非ゼロの値を含む3dテンソルがあります。ゼロ以外のすべての値をゼロに、ゼロの値を特定の値に置き換えたい。どうやってやるの？

pytorch
8
Wasi Ahmad

次のように、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]
7
entrophy

使用する

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]])

詳細はこちらをご覧ください https://pytorch.org/docs/stable/torch.html#torch.where

1
Eli Safra