我想在rgb图像中获取每个通道的最大值。 目前，我正在分别迭代每个频道：
r_max = rgb[1]:max()
g_max = rgb[2]:max()
b_max = rgb[3]:max()
有没有办法把它理解为Torch中的一个声明？有人可以帮我解决可能的解决方案吗？
答案 0 :(得分：2)
Max采用可选的＆＃39;尺寸＆＃39;论点。 在通道以外的尺寸上使用它会给你一个3x1x1张量，如果需要你可以展平，例如
rgb_max = rgb:max(2):max(3):reshape(3)