bugfix> tensorflow > 投稿

次元を減らすためのテンソルフローにAPIはありますか?たとえば、テンソル [[1,2], [3,4], [5,6]] が与えられた場合最後の次元を減らして [1, 2, 3, 4, 5, 6] にしたい 。

API tf.expand_dims() があります反対のことをします。 tf.reduce_dims() はありますか ?

回答 1 件
  • 1-D を削除できます   tf.squeeze を使用したディメンション  または、 tf.reshape を使用して、互換性のある別の形状のテンソルの形状を変更できます。

    あなたの場合、あなたは 1-D を持っていないので  寸法を変更すると、reshapeを使用して必要なものを取得できます。

    a と仮定します  入力テンソルである場合、次を使用して入力を「平坦化」できます。

    b = tf.reshape(a, (-1,))
    
    

あなたの答え