Я пытаюсь преобразовать входные данные моей нейронной сети, чтобы получить тензор следующим образом (например, размер пакета = 3):
Исходный ввод: [1,2,3],
Преобразованный ввод: ([0,1,2],[1,2,3],[2,3,0])
Я получаю исходные данные из tfrecord с помощью tf.data, и для преобразования ввода мне нужно знать истинный размер пакета, потому что последний пакет меньше. Но я использую Оценщики и не могу этого сделать:
with tf.Session() as sess:
true_batch = tf.shape(original_input)[0]
true_batch = sess.run(true_batch)
Есть ли способ сделать это внутри функции модели Оценщиков, или мне нужно изменить мой набор данных?
Спасибо