Torch:得到 RGB 图像中每个通道的最大值

我正在尝试得到 RGB 图像中每个通道的最大值。 目前,我是分别对每个通道进行迭代,如下所示:

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

是否有任何方法将此简化为 Torch 中的单个语句?是否有人能够帮助我提供可能的解决方案?

点赞
用户6858358
用户6858358

Max 函数可以使用可选的 'dimension' 参数。 如果应用于通道以外的维度,则会得到一个 3x1x1 的张量,如果需要,可以通过压缩使其变成一维张量,例如:

 rgb_max = rgb:max(2):max(3):reshape(3)
2016-09-21 12:51:22