tf.squeezeをtf.reshapeとして置き換えてください。

Question

テンソルフローを使用してモバイルネットをトレーニングする必要があります。 tf.squeezeレイヤーはサポートされていません。 tf.reshapeで置き換えることはできますか？

の動作である：ネットシェイプ[50,1,1,1000]を有する

tf.reshape(net, [50,1000], name='reshape') 

：として

tf.squeeze(net, [1, 2], name='squeeze') 

同じ。

Answer 1

なぜtf.squeezeはサポートされていませんか？テンソルから1次元軸を削除するには、tf.squeezeが正しい操作です。しかし、あなたはtf.reshapeであなたの望む仕事を達成することができますが、私はあなたにtf.squeezeを利用することを提案します。