max(1) will return the maximal value (and index in PyTorch) in this particular dimension.
Both, numpy and PyTorch return the same values (PyTorch additionally with the indices):
x = np.random.randn(10, 10)
print(x.max(1))
print(torch.from_numpy(x).max(1)[0])