pytorch max()

用法

torch.max(input, dim, keepdim=False, out=None) -> (Tensor, LongTensor)

它是一个过载函数，函数参数不同函数的功能和返回值也不同。
input是一个tensor.
当max函数中有维数参数的时候，它的返回值为两个，一个为最大值，另一个为最大值的索引.
当维度为0时，并不是对每行求值，而是对每列求值，返回shape为一行的结果。
当维度为1时，并不是对每列求值，而是对每行求值，返回的结果为一列。
如：

# 对每一行求最大值，即找到IoU最大的box
# 函数返回NXM的矩阵，对每行求最大值，即对每个pred box，求出与它IoU最大的target
ious, i = box_iou(pred[pi, :4], tbox[ti]).max(1)