Tensorflow変数の次元を増やす方法

Question

tf.embedding_lookupを実行すると、ベクトル（行列ではない）が返されます。

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,1)にこの(20,8000)テンソルを再構築する必要があります。私はどうすればいいのですか？ 私はtf.reshapeを使用してハードコード（20,8000,1）を求めていません。私は一般的に2d - > 3d以上を変換する方法を求めています。

Answer 1

tf.expand_dimsを使用できます。この操作では、テンソルの形状に1のディメンションが挿入されます。

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