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

Question

假设我们有一个张量。

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

我们如何将它转换为一个整数张量，其值为[1, 0, 1, 0]？

- Anjani Dhrangadhariya

Answer 1

解决方案只需一行代码。

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

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

- Anjani Dhrangadhariya