mcE*_*nge 7

已经找到了:

a = torch.zeros(10)
b = a.type(torch.ByteTensor)
Run Code Online (Sandbox Code Playgroud)

  • 更短的版本:`torch.zeros(10).byte()`。 (4认同)
  • Pytorch 现在有 `torch.bool`。请使用它代替“torch.ByteTensor”。 (4认同)

mid*_*das 6

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


小智 5

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

a = torch.zeros(10, dtype=torch.uint8)
Run Code Online (Sandbox Code Playgroud)

  • 现在有一个您可以(应该?)使用的 [`torch.bool`](https://pytorch.org/docs/stable/tensor_attributes.html#torch.torch.dtype) 类型。 (3认同)