Я создаю несколько тензоров в цикле и хочу объединить их все в один объект тензора тензоров.Я думаю, что мне не хватает некоторой части API-интерфейса тензорного потока scala, потому что то же самое довольно просто в python с использованием tf.concat.
Каковы эквивалентные команды для tf.concat / tf.split в scala?
Это в основном программа на python, которую я хочу скопировать в Scala (не могу найти tf.concat иtf.split для Scala):
import tensorflow as tf
# Initialize two constants
x1 = tf.constant([1,2,3,4])
x2 = tf.constant([5,6,7,8])
x11 = tf.constant([4,3,2,1])
x22 = tf.constant([8,7,6,5])
# concatenate x1, x11 and x2, x22 tensors
x111 = tf.concat([x1, x11], 0)
x222 = tf.concat([x2, x22], 0)
# Multiply
result_bat = tf.multiply(x111, x222)
# Intialize the Session
sess = tf.Session()
# Print the result
res = sess.run(result_bat)
res10, res11 = tf.split(res, 2, 0)
print(sess.run(res10))
print(sess.run(res11))
# Close the session
sess.close()