标签: pytorch
我想创建一个仅包含布尔值的张量。 在Matlab中将是
a = false(10,1)
答案 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)