我有一批形状为?,192的张量。现在我想从这批中获取任何单个随机张量。所得张量应具有1,192的形状,而不是应在Keras中完成。对如何完成这项工作有任何想法吗?
?,192
1,192
该问题通过带有lambda函数的张量流操作解决
Var= Lambda(lambda inputs: tf.gather_nd(inputs, [[0]]) )(Var)