Как переписать тензорные потоки tf.reduce_sum в python, используя операции линейной алгебры?

Question

Я пытаюсь переписать операцию tf.reduce_sum для

tf.reduce_sum(tf.square(self.kernel), [0,1,2], keepdims=False)

Использовать только основные операции линейной алгебры. Контекст для этого заключается в том, что я пытаюсь преобразовать модель в tenorrt, которая в настоящее время не поддерживает операцию redu_sum.

Поскольку оператор redu_sum представляет собой просто линейную операцию, должен быть способ написать ее, используя гораздо более простые операции. К сожалению, я не достаточно квалифицирован в линейной алгебре или тензорном потоке, чтобы сделать это самостоятельно и надеюсь получить некоторую помощь.

Заранее большое спасибо!

Answer 1

результат = numpy.sum (square_output_tensor.numpy (), ось = (0,1,2), keepdims = False)

Предполагается, что вы работаете в активном режиме, иначе вызов метода .numpy () не будет работать. Вместо этого для режима графика вы бы использовали tf.py_func.