将PyTorch张量值获取到Python集中

Question

我有一组表示图像蒙版的pytorch张量（形状为(1, x, x)），我没有标签，需要从这个pytorch张量中手动获取它们。

有没有一种有效的方法可以将这些值放入集合中而不必遍历每个单元格。