Каналы отображения тензорного потока при сохранении другого измерения

Question

Я пытаюсь сопоставить набор активаций X размера (?, 200, 300, 2000)

представлению (?, 200, 300, 100).

Для этого у меня есть весовая матрица W размера (2000, 100),Как я могу добиться, чтобы каждый (?, x1, y1, 2000) умножался правильно?Я пробовал tf.matmul и tf.tensordot, но не смог заставить его работать.

Answer 1

Это должно помочь вам (переходя к более мелким фигурам для ускорения вычислений):

X = np.ones(((5, 2, 3, 7)))
W = np.ones((X.shape[3], 10))
X_reshaped = tf.reshape(X, [-1, X.shape[3]])
# Shape: (30, 7)
y = tf.matmul(X_reshaped, W)
# Shape: (30, 10)
y_reshaped = tf.reshape(y, [-1, X.shape[1], X.shape[2], 10])
# Shape: (5, 2, 3, 10)