使用tf.split機能(tensorflow1.13.0)

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]

おすすめ

転載: www.cnblogs.com/HYWZ36/p/11408451.html