如何在 PyTorch 中将布尔值张量转换为整数?

How to convert a tensor of booleans to ints in PyTorch?

假设,我们有一个张量

t = torch.tensor([True, False, True, False])

我们如何将其转换为值为 [1, 0, 1, 0] 的整数张量?

只需一行代码即可解决。

要将值为 [True, False, True, False] 的张量 t 转换为整数张量,只需执行以下操作。

t = torch.tensor([True, False, True, False])
t_integer = t.long()
print(t_integer)
[1, 0, 1, 0]