2017-10-23 13 views
0

私はtf.batch_to_space_ndがこの作業に必要なものだと信じていますが、このインデックス作成とシェイプ/ランクビジネス。[n * 7,12]を[n、7,12]にバッチ処理する方法

私がしたいのは、形状[n * 7、12]の入力テンソルから最初の7行を取り出して、別の次元に入れたいということです。次の7行などを取るよりも、最終的な形状は[n、7,12]である必要があります。

おそらく、単に次元を追加または絞り込むことができます。入力は[n * 7、12、1、1]、出力は[n、7、12、1]となります。私はこれがbatch_to_space_ndか、おそらくtf.gather_ndや他のslice and join操作のどれかで実行できますが、正しい組み合わせが見つからないと思います。

本当にスライスしてスタックする必要はありますか?私はそれを行う方法を知っていますが、不必要に醜いようです。

答えて

1

単純にtf.reshapeを使用するように見えます。

# Assuming x has shape [n*7, 12], y will have shape [n, 7, 12] 
y = tf.reshape(x, [n, 7, 12]) 
関連する問題