keras tensorflow 中的前向插补

juud5qan 于 2022-11-13 发布在其他

关注(0)|答案(2)|浏览(231)

作为一个最小的例子，假设我有一个Tensor的形式：

[[ 1.   0.   3. ]
 [ 7.   5.   6. ]
 [ 0.   0.   0. ]
 [ 0.  11.   1. ]
 [ 0.   0.   0. ]
 [ 0.   0.   0. ]
 [13.  14.  16.5]]

是否有一种方法（在tensorflow中原生）插补完全归零的行，使其值等于最后一个非完全归零的行？即-〉：

[[ 1.   0.   3. ]
 [ 7.   5.   6. ]
 [ 7.   5.   6. ]
 [ 0.  11.   1. ]
 [ 0.  11.   1. ]
 [ 0.  11.   1. ]
 [13.  14.  16.5]]

我曾考虑过使用tf.tensor_scatter_nd_update，但没有成功。

keras

来源：https://stackoverflow.com/questions/74286259/forward-imputation-in-tensorflow

2条答案

按热度按时间

5rgfhyps1#

我们可以使用tf.gather(a, indices)来获得上面的输出。
indices必须是[0, 1, 1, 3, 3, 3, 6]，可以使用下列程式码取得：
第一个

赞(0）回复(0）举报 2022-11-13

vql8enpb2#

这段代码也可以在GPU上运行。
第一个
输出量：

<tf.Tensor: shape=(7, 3), dtype=float32, numpy=
array([[ 1. ,  0. ,  3. ],
       [ 7. ,  5. ,  6. ],
       [ 7. ,  5. ,  6. ],
       [ 0. , 11. ,  1. ],
       [ 0. , 11. ,  1. ],
       [ 0. , 11. ,  1. ],
       [13. , 14. , 16.5]], dtype=float32)>

赞(0）回复(0）举报 2022-11-13

我来回答

keras tensorflow 中的前向插补

2条答案

相关问题

热门标签

最新问答