Tensorflow如何增加變量的維數

當我們做tf.embedding_lookup時，它返回一個向量（不是矩陣）。Tensorflow如何增加變量的維數

In [244]: one_hot_label = tf.nn.embedding_lookup(np.eye(vocab_size), Y[labels_i]) 

In [245]: one_hot_label 
Out[245]: <tf.Tensor 'embedding_lookup_43975:0' shape=(20, 8000) dtype=float64>

我需要這個(20,8000)張重塑成(20,8000,1)。我應該怎麼做？我沒有要求使用tf.reshape的硬線（20,8000,1）。我一般會問如何轉換2d - > 3d或更高。

來源

2017-09-27 Aaron

您可以使用tf.expand_dims：此操作將1的維度插入到張量的形狀中。

one_hot_label = tf.expand_dims(one_hot_label, axis=2)

來源

2017-09-27 08:05:59 nessuno

Tensorflow如何增加變量的維數

回答

相關問題