Solved by David_Reiss in post #2 
              
              
                No, and we have no plans to add it.  Please update your TorchScript model to do any tensor operations you need so that the Java code can simply provide raw input and consume raw output.