How to get predict result from model's forward in a number

1

Here is my model:

class Model(nn.Module):
	def __init__(self):
		self.layer = nn.Sequential(
            nn.Linear(100, 100),
            nn.Tanh(),
            nn.LogSoftmax()
        )
		
	def forward(self, arg):
		return self.layer(arg)
		
model = Model()
output = model(arg)

The output is FloatTensor.
For example, with the MNIST task, how can I convery the output into a number?

2 
torch.max(output, 1)[1]