torch中获取一个一维BoolTensor中所有True的位置

可以使用torch.nonzero()函数来获取一个一维BoolTensor中所有True的位置，示例如下：

import torch

x = torch.BoolTensor([True, False, True, True, False])
indices = torch.nonzero(x)
print(indices)

输出结果为：

tensor([[0],
        [2],
        [3]])

其中，`indices`就是一个二维的tensor，表示所有True的位置。