What do you mean by “multioutput classification”?
Currently your code returns logits of dimension [batch_size, 10].
So it should work, if you have 10 classes.
Could you explain a bit more about, what doesn’t work with the code?
If the problem lies in the model input, you should change input_dim to 240.