2017-07-26 8 views
2

テンソルは単純にベクトルで、vector = [0.5 0.4]であり、tf.shapeはshape =(1、)であることを示しています。ベクトルをm回複製し、[m、2]の形をしたいと思いますm = 2、matrix = [[0.5 0.4], [0.5 0.4]]。 tf.tileを使ってどのように達成するのですか?tf.tileを使って行テンソルを複製するか?

答えて

3

次のようにすると、vecはベクトルで、乗算はm、vecを繰り返す回数です。ベクトルに対してタイルが実行され、結果が所望の構造に再形成される。

import tensorflow as tf 

vec = tf.constant([1, 2, 3, 4]) 
multiply = tf.constant([3]) 

matrix = tf.reshape(tf.tile(vec, multiply), [ multiply[0], tf.shape(vec)[0]]) 
with tf.Session() as sess: 
    print(sess.run([matrix])) 

これは、その結果:

[array([[1, 2, 3, 4], 
     [1, 2, 3, 4], 
     [1, 2, 3, 4]], dtype=int32)] 
関連する問題