一、简介 torch.nonzero是PyTorch库中的一个非常重要的函数，用于返回输入张量中非零元素的索引（对应的坐标），以便于其他操作的进行。该函数接受的输入类型可以是张量、…