手电筒：rgb 图像中每个通道的最大值
Torch: max value of each channel in an rgb image
我正在尝试获取 rgb 图像中每个通道的最大值。
目前，我像这样分别遍历每个频道：
r_max = rgb[1]:max()
g_max = rgb[2]:max()
b_max = rgb[3]:max()
有没有办法在 Torch 中将其理解为一条语句？有人可以帮我解决可能的问题吗？
Max 采用可选的 'dimension' 参数。
在通道以外的维度上使用它会给你一个 3x1x1 张量，你可以根据需要将其展平，例如
rgb_max = rgb:max(2):max(3):reshape(3)
我正在尝试获取 rgb 图像中每个通道的最大值。 目前，我像这样分别遍历每个频道：
r_max = rgb[1]:max()
g_max = rgb[2]:max()
b_max = rgb[3]:max()
有没有办法在 Torch 中将其理解为一条语句？有人可以帮我解决可能的问题吗？
Max 采用可选的 'dimension' 参数。 在通道以外的维度上使用它会给你一个 3x1x1 张量，你可以根据需要将其展平，例如
rgb_max = rgb:max(2):max(3):reshape(3)