Here is the code that I wanna implement in java for the PyTorch mobile application, but there is seem like no related API.
pred = torch.argmax(outputs, 1) pred = pred.cpu().data.numpy() predict = pred.squeeze(0)
final IValue outputTuple = module.forward(IValue.from(inputTensor)).toTuple(); final Tensor outputTensor = outputTuple.toTensor(); Log.d(TAG, "onCreate: " + outputTensor.getDataAsFloatArray());
The FloatArray is 1-dim array, it really rack my brains, Could anyone please tell me a clue. thank you