tf.split(入力、num_split、寸法):
寸法入力は、それが0である場合、それは第一の切断寸法ことを意味し、一次元、0テンソルを意味し; num_split番号が切断され、それが2を表すが、入力テンソルを2つの部分に切断されている場合、それぞれがリストです。
例えば:
TFとして輸入tensorflow。 NPとしてnumpyのインポート。 A = [1,2,3]、[4,5,6]] X = tf.split(A、3,1) のSESとしてtf.Session()を持つ: C = sess.run(X) ELE用C中: プリント(ELE)
出力:
[1]
[4]
[2]
[5]
[3]
[6]