Как реализовать эту операцию в TensorFlow?

Question

Предположим, у нас есть два тензора X, Y форм, скажем, (a, b) и (b, c) соответственно. Затем мы можем построить новый тензор T формы (a, b, c), определенный как T[i][j][k] = X[i][j] + Y[j][k].

Как я могу реализовать это в TensorFlow? Как насчет аналога, где я заменяю сложение умножением?

Заранее спасибо.

Answer 1

Это то, что вы ищете?

a = tf.constant([[2,3],[4,1]]) # A 2x2 matrix
b = tf.constant([[1,2,1],[3,4,2]]) # A 2x3 matrix

# Make a into 2x2x3 matrix
aa = tf.expand_dims(a,-1)
# Make b into 1x2x3 matrix
bb = tf.expand_dims(b, 0)

# Do element wise addition, resulting in 2x2x3 matrix.
c = aa + bb