Pytorch:创建一个布尔张量(类型:torch.ByteTensor)?

Pytorch: Create an boolean tensor (type: torch.ByteTensor)?

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

a = false(10,1)

已经找到:

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

这不是更经济(虽然更长):

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

或者,在旧版本中,

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

(感谢@drevicko 提供指向 bool 的指针。)