如何在Pytorch中反转布尔值的张量?

时间:2019-12-03 01:53:29

标签: python pytorch

使用NumPy,您可以使用np.invert(array)来实现,但是Pytorch中没有invert函数。假设我有一个布尔值的二维张量:

import torch

ts = torch.rand((10, 4)) < .5
print(ts)
Out[33]: 
tensor([[ True,  True, False,  True],
        [ True,  True,  True,  True],
        [ True, False,  True,  True],
        [False,  True,  True, False],
        [False,  True,  True,  True],
        [ True,  True,  True,  True],
        [ True, False,  True,  True],
        [False,  True, False,  True],
        [ True,  True, False,  True],
        [False, False,  True, False]])

如何将False转换为True,反之亦然?

1 个答案:

答案 0 :(得分:0)

从字面上看,只需使用波浪号将所有True转换为False,反之亦然。

ts = ~ts