Keras-如何从一批张量中仅获得一个张量?

问题描述 投票:0回答:1

我有一批形状为?,192的张量。现在我想从这批中获取任何单个随机张量。所得张量应具有1,192的形状,而不是应在Keras中完成。对如何完成这项工作有任何想法吗?

python keras deep-learning
1个回答
0
投票

该问题通过带有lambda函数的张量流操作解决

Var= Lambda(lambda inputs: tf.gather_nd(inputs, [[0]]) )(Var)
© www.soinside.com 2019 - 2024. All rights reserved.