こんにちは、テンソルフローは初めてです。tensorflow で次の python コードを実装したいと考えています。
import numpy as np
a = np.array([1,2,3,4,5,6,7,9,0])
print(a) ## [1 2 3 4 5 6 7 9 0]
print(a.shape) ## (9,)
b = a[:, np.newaxis] ### want to write this in tensorflow.
print(b.shape) ## (9,1)
対応するコマンドはtf.newaxis(またはNonenumpy のように) です。tensorflow のドキュメントには独自のエントリはありませんが、 のドキュメント ページで簡単に言及されていますtf.stride_slice。
x = tf.ones((10,10,10))
y = x[:, tf.newaxis] # or y = x [:, None]
print(y.shape)
# prints (10, 1, 10, 10)
使用tf.expand_dimsも問題ありませんが、上記のリンクに記載されているように、
これらのインターフェースはより使いやすく、強く推奨されます。