Pytorch：创建一个布尔张量（类型：torch.ByteTensor）？

Question

我想创建一个只包含布尔值的张量。 在 Matlab 中，这将是

a = false(10,1)

Answer 1

这不是更经济（虽然更长）：

a = torch.zeros(10, dtype=torch.bool)

或者，在旧版本中，

a = torch.zeros(10, dtype=torch.uint8)

（感谢@drevicko 提供指向 bool 的指针。）

Answer 2

已经找到了：

a = torch.zeros(10)
b = a.type(torch.ByteTensor)

Answer 3

从版本1.2.0开始，PyTorch支持布尔张量 。