[pytorch] expose IValue for custom input data type (#1195)

Change-Id: Ifdbf35297073d9f105e6819bd632c77df4782940

.gitignore

@@ -14,6 +14,7 @@ libs
 *.dylib
 *.dll
 *.class
+*.log
 
 # Eclipse
 .settings

pytorch/pytorch-engine/build.gradle

@@ -27,11 +27,11 @@ processResources {
         ]
         def classesDir = "${project.buildDir}/jnilib"
         files.each { entry ->
-            project.logger.lifecycle("Downloading ${url}/${entry.key}")
             def file = new File("${classesDir}/${entry.value}")
             if (file.exists()) {
                 project.logger.lifecycle("prebuilt or cached file found for ${entry.value}")
             } else {
+                project.logger.lifecycle("Downloading ${url}/${entry.key}")
                 file.getParentFile().mkdirs()
                 new URL("${url}/${entry.key}").withInputStream { i -> file.withOutputStream { it << i } }
             }

pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtSymbolBlock.java

@@ -19,6 +19,7 @@
 import ai.djl.ndarray.types.Shape;
 import ai.djl.nn.AbstractSymbolBlock;
 import ai.djl.nn.SymbolBlock;
+import ai.djl.pytorch.jni.IValue;
 import ai.djl.pytorch.jni.IValueUtils;
 import ai.djl.pytorch.jni.JniUtils;
 import ai.djl.training.ParameterStore;
@@ -88,6 +89,16 @@ public void close() {
         }
     }
 
+    /**
+     * Runs the forward of this PyTorch module.
+     *
+     * @param inputs the input {@link IValue}
+     * @return the result {@link IValue}
+     */
+    public IValue forward(IValue... inputs) {
+        return IValueUtils.forward(this, inputs);
+    }
+
     /** {@inheritDoc} */
     @Override
     protected NDList forwardInternal(