TensorFlow 总结给定索引

Question

我正在尝试获取两个矩阵的差异，以便我可以定义我的损失函数。模型很简单，我有一个输入矩阵和一个输出矩阵。

将 X 定义为输入矩阵

定义Y为输出矩阵

通常我会做一个tf.reduce_mean(tf.abs(X-Y))，但这是不可能的，因为矩阵 X 包含 nan 值。

所以我想要做的是将 X 相加，其中False 由tf.is_nan(X) 给出，然后我将在与 X 相加的相同索引处相加 Y。然后我将定义我的损失为 loss = tf.abs(reduce_nan_sum(X)-reduce_nan_sum(Y))

with tf.Session() as sess:
    sess.run(init)
    print(sess.run(tf.is_nan(X), feed_dict={X: vals}))

[[False, False,  True],
 [False,  True, False]]

样本 X 和 Y 值

X = [[0.,  1.,  nan],
    [2.,  nan, 0.5]]

Y = [[0.002,  0.967,  0,2],
    [1.956,  0.3, 0.487]] 

Answer 1

NaN 可以用零替换以进行操作。使用与Replace nan values in tensorflow tensor相同的解决方案

sess.run(tf.reduce_mean(tf.abs(tf.where(tf.is_nan(x), tf.zeros_like(x), x)-y)))