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 的指针。)
我想创建一个只包含布尔值的张量。 在 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 的指针。)