Как вычислить потерю движителя Земли на 2D-массивах в Керасе?

Question

Я искал, как реализовать потерю движителя Земли на 2d массивах в Керасе. Мой ввод - 2d (не изображения). Я пробовал этот код ниже из https://github.com/master/nima/blob/4a685993d0e5942cf70de54f1c31a218827ccba3/nima.py#L31 Но, похоже, он не работает. Кто-нибудь может с этим помочь?

          def ecdf(p):
       n = p.get_shape().as_list()[1]
       indices = tril_indices(n)
       indices = tf.transpose(tf.stack([indices[1], indices[0]]))
       ones = tf.ones([n * (n + 1) / 2])
       triang = tf.scatter_nd(indices, ones, [n, n])
       return tf.matmul(p, triang)

    def emd_loss(p, p_hat, r=2, scope=None):
      with tf.name_scope(scope, 'EmdLoss', [p, p_hat]):
      ecdf_p = ecdf(p)
      ecdf_p_hat = ecdf(p_hat)
      emd = tf.reduce_mean(tf.pow(tf.abs(ecdf_p - ecdf_p_hat), r), 
      axis=-1)
      emd = tf.pow(emd, 1 / r)
      return tf.reduce_mean(emd)