用于张量乘法的 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.