我想做这样的事情.
假设我们有一个张量A.
A = [[1,0],[0,4]]
我希望从中获得非零值及其指数.
Nonzero values: [1,4]
Nonzero indices: [[0,0],[1,1]]
Numpy也有类似的操作.根据非零指数
np.flatnonzero(A),在展平的A.
x.ravel()[np.flatnonzero(x)]extract元素中返回非零的索引.
这是这些操作的链接.
我怎样才能在Tensorflow中使用python执行上面的Numpy操作？
(矩阵是否扁平化并不重要.)