Tensordot 在 Tensorflow 中的使用

问题描述

在我的 python Tensorflow 代码中,我有一个维度为 (201,25,25) 的 3d 张量 "a" 和维度为 (201,25) 的二维张量 "b",我想计算乘积 a*b 其中结果应在其第 i 行中具有使用张量点方法计算的以下乘积。

tf.tensordot(a[i,:,:],b[i,1) // This product has dim of (25,) 有没有一种使用 tensordot 函数一次计算这个乘积的好方法,我曾尝试这样做,但我做不到。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)