Я пытаюсь сделать matmul из двух тензоров, один - [None, 4, 256], а другой - его транспонирование (я использую tf.transpose и получил тензор [256, 4, None]). Ожидаемый результат - тензор [4, 4]. Когда я использую matmul, он возвращает ошибку. Мне интересно, как я могу получить ожидаемый результат? Спасибо!
#inputs: [None, 4, 256]
inputs_transpose = tf.transpose(inputs, perm = [0, 2, 1]) #[None, 256, 4]
temp_weights = tf.matmul(inputs, inputs_transpose) #[4, 4]expected