indices by element, rather than dimension, use:
transpose(nonzero(a))
The result of this is always a 2-...简单来说就是参数是数组或者矩阵,返回值为该数组或者矩阵中非零元素的下标值构成的元组。该元组有两维,第一维是非零元素所在的行,第二维是非零元素所在的列。...[0,0,0]])
则 nonzero(a) 返回值为 (array([0, 1]), array([0, 0])) , 因为矩阵a只有两个非零值, 在第...0行、第0列,和第1行、第0列。...所以结果元组中,第一个行维度数据为(0,1) 元组第二个列维度都为(0,0)。