标签: tensorflow
给定形状A的张量[?, n, l]和形状D的张量[?, n],我想要分割形状a的每一行{ [n,l] A D的相应标量{1}}。{/}
A
[?, n, l]
D
[?, n]
a
[n,l]
我以为我可以通过使用tf.div的广播行为以某种方式做到这一点。不幸的是我被困住了。
tf.div
答案 0 :(得分:1)
在这里,您需要扩展D以匹配A的维度:
res = A / D[...,tf.newaxis])