使用批处理数据进行逐行划分

时间:2017-08-05 13:13:02

标签: tensorflow

给定形状A的张量[?, n, l]和形状D的张量[?, n],我想要分割形状a的每一行{ [n,l] A D的相应标量{1}}。{/}

我以为我可以通过使用tf.div的广播行为以某种方式做到这一点。不幸的是我被困住了。

1 个答案:

答案 0 :(得分:1)

在这里,您需要扩展D以匹配A的维度:

res = A / D[...,tf.newaxis])