pytorch torch.max返回2个值

今早调用torch.max的时候出现了tuple,
后来发现是因为它返回2个值

请作者喝一杯咖啡☕️