テンソルフローの合計プール

Question

テンソルフローのsumプーリングの既存の実装はありますか？ documentationを検索すると、平均および最大プーリング操作のみがサポートされているようです。

私はNHWCテンソルを持っています。ここで、各HxWマトリックスは2x2サムプーリングを介してダウンサンプリングしたい確率マップです。もし、集約関数がテンソルフローに存在しない場合は、Python APIを使ってこのダウンサンプリング演算を実装する別の方法がありますか？

Answer 1

現時点では、テンソルフローにそのようなものはありません。次の最良のものは、平均プーリングの出力に4を掛けるだけです。要素ごとの乗算は、この追加されたステップについて心配する必要はありません。

あなたdoしかし、プロセスの精度のビットのペア。

代わりに、自分で合計を計算することもできます。

y = (x[:,0::2,0::2,:] + 
    x[:,1::2,0::2,:] + 
    x[:,0::2,1::2,:] + 
    x[:,1::2,1::2,:]) 

これは以前の解決策よりも遅くする必要がありますが、それが重要な場合は精度が若干向上する可能性があります。

Answer 2

ユーザーは、平均プールを利用するために平均プールを利用できることを既に教えています。彼はまたあなたが精度を失うかもしれないと言いました（しかし、これはほとんど起こりそうもなく、ほとんどの場合無関係です）。

もう1つの簡単な方法があります。あなたは2d畳み込みとストライドを使って合計プールを計算することができます。あなたのカーネルは1の行列です。