У меня есть набор тезоров Pytorch целых чисел, представляющих маски изображений (форма (1, x, x)), у меня нет меток, и мне нужно вручную извлечь их из этого тензора Pytorch.
Существует ли эффективный способ поместить эти значения в набор без необходимости перебирать каждую ячейку.