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

时间:2018-01-08 13:48:04

标签: pytorch

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

a = false(10,1)

3 个答案:

答案 0 :(得分:6)

从1.2.0版开始,PyTorch支持布尔值tensors

答案 1 :(得分:3)

已经找到了:

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

答案 2 :(得分:2)

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

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