我有一个张量'a'，形状为（无，85,1500）。索引张量'b'具有形状（无，5）。索引张量用于在轴= 1的'a'中选择值（该值为85）。预期结果的形状为（None，5,1500）。
但是当我使用Order.aggregate({
$group: {
_id: '',
totalAmount: { $sum: '$total' }
}
}, {
$project: {
_id: 0
}
})
时。我的状态为（无，5,85,1500）。
使用tf.gather (a,b)时，我得到了（无，无，5,1500）
使用tf.gather (a,b,axis=1)时出现错误：
tf.gather_nd(a,b,)有人可以帮我吗？谢谢。