My model need run torch.Tensor.max() meth to get the final index of my word dictionaries.
But now I don’t know what to do.
Oh, I found I could write one myself:sweat_smile:
But hopefully the mobile.Tensor will add some math util:laughing:
We are not planning to implement the full tensor API in the Java bindings. If you want some extra result calculated (like the max), your best bet is to compute it as part of your model.