火炬：在RGB图像中的每个通道的最大值

问题描述：

我想获得每个通道在RGB图像中的最大值。目前，我分别遍历每个通道是这样的：火炬：在RGB图像中的每个通道的最大值

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

有什么办法来理解成火炬单个语句呢？有人可以帮助我解决可能的问题吗？

不好得多，但你可以移动到一个行：r_max时，g_max，b_max = RGB [1]：MAX（），RGB [2]：MAX（），RGB [3]：MAX（） –

答

Max采用可选的“维度”参数。在通道以外的尺寸上使用它会给你一个3x1x1的张量，如果需要你可以将它变平，例如，

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