一、簡介 torch.nonzero是PyTorch庫中的一個非常重要的函數，用於返回輸入張量中非零元素的索引（對應的坐標），以便於其他操作的進行。該函數接受的輸入類型可以是張量、…