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

Question

假设，我们有一个张量

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

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

Answer 1

解决方案只是一行代码。

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

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