pytorch--BoolTensor进行逻辑运算操作_torch两个bool类型的tensor做运算

作者：小丑西瓜9 | 2024-03-18 14:14:57

踩

torch两个bool类型的tensor做运算

since pytorch 1.2, there is a dtype torch.bool and associated BoolTensor. The bitwise operators aka numpy (and &, or |, xor ^ and not ~) all work on these as logical operations.

BoolTensor在pytorch1.2后引入，针对BoolTensor才有了如下逻辑运算（dtype为bool型）

与操作：a & b
或操作：a | b
异或操作：a ^ b
取反操作：~a

参考自Implementing element-wise logical and tensor operation

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小丑西瓜9/article/detail/263627