nonzero

该OP返回输入中非零元素的坐标。如果输入 x 有 n 维，共包含 z 个非零元素，当 as_tuple = False 时，返回结果是一个 shape 等于的 Tensor , 第 i 行代表输入中第 i 个非零元素的坐标；当 as_tuple = True 时，返回结果是由 n 个大小为的 1-D Tensor 构成的元组，第 i 个 1-D Tensor 记录输入的非零元素在第 i 维的坐标。