Hi tom, yes, torch.max can give the right output and it can also give the maximum value of all the elements of a tensor. But I was wondering why torch.maximum would not work, since it’s pretty clear this function is designed for it as in https://pytorch.org/docs/master/generated/torch.maximum.html#torch.maximum 153