用於張量乘法的 keras/tensorflow 廣播
[英]keras/tensorflow broadcast for tensor multiplication
問題描述
我有兩個張量：張量a形狀為(n, 1) ，張量b形狀為(p, q) ，其中p和q都很小，但n很大。 我想將他們的“外積”張量c作為形狀為(n, p, q)的張量，這樣它的元素是
c[i, j, k] = a[i, 0] * b[j, k]
我應該如何使用 tensorflow/keras 廣播或函數來實現這一點而不會產生循環？
1 個解決方案
解決方案1
1 已采納 2020-02-28 13:46:31
你可以簡單地做：
c = tf.expand_dims(a, 2) * b
在tensorflow中的tensordot操作中有沒有辦法廣播張量？
[英]is there a way to broadcast tensor in tensordot operation in tensorflow?
'Tensor'對象沒有屬性'_keras_history'Keras沒有Tensorflow張量
[英]'Tensor' object has no attribute '_keras_history' Keras with no Tensorflow tensor
聲明:本站的技術帖子網頁，遵循CC BY-SA 4.0協議，如果您需要轉載，請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.