керас / тензор потока трансляции для тензорного умножения - PullRequest
1 голос
/ 28 февраля 2020

У меня есть два тензора: тензор a имеет форму (n, 1), а тензор b имеет форму (p, q), где p и q малы, но n велики. Я хотел бы получить их тензор "внешнего произведения" c в качестве тензора с формой (n, p, q), такой, чтобы его элементы были

c[i, j, k] = a[i, 0] * b[j, k]

Как использовать тензор потока / керас трансляция или функции для достижения этой цели, не неся при этом все oop?

1 Ответ

1 голос
/ 28 февраля 2020

Вы можете просто сделать:

c = tf.expand_dims(a, 2) * b
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...