Тензор потока Einsum получает след неправильно?

Question

Почему эти выходы отличаются?Это намеренное поведение?Я использую tenorflow 1.12

import tensorflow as tf

matrix = tf.constant([[1, 2, 3],
                      [4, 5, 6],
                      [7, 8, 9]])

with tf.Session() as sess:
    print(tf.einsum('ii',matrix).eval())

# 45

, в то время как

import numpy as np

np.einsum('ii',[[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])
# 15 (as it should be)

Почему эйнсум tenorflow суммирует все элементы матрицы?

ОБНОВЛЕНИЕ: Я открылпроблема на GitHub, похоже на ошибку.

Answer 1

Это было исправлено и объединено в PR 25079.