Pytorch深度学习随手记（3）np.argwhere()用法

np.argwhere( a )

返回非0的数组元组的索引，其中a是要索引数组的条件。

是不是可以理解为去掉0项的数组的下标为返回值？

默认按列返回，可以再操作一下按行返回

比如说：

 1      1      0      1      1      0      1      2      0      1      1      4      1      0      0      8        6

0       1      2      3      4      5      6      7      8      9     10    11    12    13    14    15      16