Torch：rgbイメージ内の各チャネルの最大値

Question

rgbイメージ内の各チャネルの最大値を取得しようとしています。 現在、私はこのような個別チャネルを反復処理しています：

r_max = rgb[1]:max() 
g_max = rgb[2]:max() 
b_max = rgb[3]:max() 

はトーチ1つの文にこれを理解する方法はありますか？誰かが可能な解決策で私を助けてくれますか？

Answer 1

Maxはオプションの 'dimension'引数をとります。 チャネル以外の次元でこれを使用すると、3x1x1のテンソルが得られます。必要に応じて平坦化することができます。

rgb_max = rgb:max(2):max(3):reshape(3)