python - 属性错误 : 'torch.return_types.max' object has no attribute 'dim' - Maxpooling Channel

Question

我正在尝试对 channel 维度进行 maxpooling:

class ChannelPool(nn.Module):
    def forward(self, input):
        return torch.max(input, dim=1)

但我得到了错误

AttributeError: 'torch.return_types.max' object has no attribute 'dim'

Answer 1

torch.max 函数返回一个元组，所以:

class ChannelPool(nn.Module):
    def forward(self, input):
        input_max, input_indexes = torch.max(input, dim=1)
        return input_max