如何在 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]