How to convert a tensor of booleans to ints in PyTorch ...
stackoverflow.com › questions › 62150659Jun 02, 2020 · The solution is just a single line of code. To convert a tensor t with values [True, False, True, False] to an integer tensor, just do the following. t = torch.tensor ( [True, False, True, False]) t_integer = t.long () print (t_integer) [1, 0, 1, 0] Share. Improve this answer. Follow this answer to receive notifications. edited May 12 at 14:57.
torch.Tensor — PyTorch 1.10.1 documentation
https://pytorch.org/docs/stable/tensorstorch.ByteTensor. /. 1. Sometimes referred to as binary16: uses 1 sign, 5 exponent, and 10 significand bits. Useful when precision is important at the expense of range. 2. Sometimes referred to as Brain Floating Point: uses 1 sign, 8 exponent, and 7 significand bits. Useful when range is important, since it has the same number of exponent bits ...
torch.masked_select — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.masked_select. Returns a new 1-D tensor which indexes the input tensor according to the boolean mask mask which is a BoolTensor. The shapes of the mask tensor and the input tensor don’t need to match, but they must be broadcastable. input ( Tensor) – the input tensor. out ( Tensor, optional) – the output tensor.