Как выбрать случайные записи из двух рваных тензоров? Например,
c = tf.ragged.constant([[1, 2, 3], [4, 5]])
v = tf.ragged.constant([[10., 20., 30.], [40., 50.]])
r = tf.random.uniform([1, 1], maxval=2, dtype=tf.int32)
with tf.Session() as sess:
print(sess.run([tf.gather_nd(c, r), tf.gather_nd(v, r)]))
дает мне либо [1, 2, 3]
и [10., 20., 30.]
, либо [4, 5]
и [40., 50.]
. Но теперь я хочу выбрать случайное число между 0 и длиной возвращаемого списка и получить соответствующие записи из обоих списков. И затем я хочу пакетировать весь этот процесс.
Заранее спасибо за вашу помощь!