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

Pytorch深度学习随手记(3)np.argwhere()用法_第1张图片

np.argwhere( a )

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

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

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

比如说:

Pytorch深度学习随手记(3)np.argwhere()用法_第2张图片

 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

你可能感兴趣的:(从入门到放弃,pytorch,深度学习,人工智能)