Керас - Как получить только один тензор из партии тензоров?

Question

У меня есть партия тензоров формы ?,192. Теперь я хочу получить любой случайный тензор из этой партии. результирующий тензор должен иметь форму 1,192, обратите внимание, что это должно быть сделано в Керасе. Есть идеи, как это сделать?

Answer 1

Проблема была решена с помощью тензорного потока с лямбда-функцией

Var= Lambda(lambda inputs: tf.gather_nd(inputs, [[0]]) )(Var)