テンソルデータのスライス操作を実現します。
843 ワード
以下の通りです
import tensorflow as tf
a=tf.constant([[[1,2,3,4],[4,5,6,7],[7,8,9,10]],
[[11,12,13,14],[20,21,22,23],[15,16,17,18]]])
print(a.shape)
b,c=tf.split(a,2,0) # 1、 2、 3、 b,c
print(b.shape)
print(c.shape
with tf.Session() as sess: #
print(sess.run(b))
print(sess.run(c))
出力結果は
(2, 3, 4)
(1, 3, 4)
(1, 3, 4)
[[[ 1 2 3 4]
[ 4 5 6 7]
[ 7 8 9 10]]]
[[[11 12 13 14]
[20 21 22 23]
[15 16 17 18]]]
このとき、bは3次元テンソルデータであり、2次元配列に変換するには、tf.rehapeコマンドが使用されます。
d=tf.reshape(b,[3,4])
print(d.shape)
#output
(3, 4)
以上のtenssorflowはテンソルデータのカット操作方式を実現しました。つまり、小編集は皆さんに全部の内容を共有しています。参考にしてもらいたいです。どうぞよろしくお願いします。