Fixed bugs with the JIT types

Author: interesaaat

TorchSharp/JIT/Module.cs

@@ -143,9 +143,13 @@ private Type GetType(Type type)
             switch (type.Kind)
             {
                 case Type.TypeKind.DynamicType:
-                    return type.AsDynamicType();
+                    var dynamic = type.AsDynamicType();
+                    type.Dispose();
+                    return dynamic;
                 case Type.TypeKind.TensorType:
-                    return type.AsDynamicType();
+                    var tensor = type.AsTensorType();
+                    type.Dispose();
+                    return tensor;
                 default:
                     return type;
             }

TorchSharp/JIT/Type/DynamicType .cs

@@ -8,12 +8,5 @@ internal DynamicType(IntPtr handle) : base(handle)
         {
             this.handle = new HType(handle, true);
         }
-
-        internal DynamicType(Type type) : base()
-        {
-            handle = type.handle;
-            type.handle = new HType(IntPtr.Zero, true);
-            type.Dispose();
-        }
     }
 }

TorchSharp/JIT/Type/Type.cs

@@ -78,21 +78,24 @@ protected void Dispose(bool disposing)
         }
 
         [DllImport("libTorchSharp")]
-        extern static short THSJIT_typeKind(HType handle);
+        extern static sbyte THSJIT_typeKind(HType handle);
 
         internal TypeKind Kind
         {
             get { return (TypeKind)THSJIT_typeKind(handle); }
         }
 
+        [DllImport("libTorchSharp")]
+        extern static IntPtr THSJIT_typeCast(HType module);
+
         internal TensorType AsTensorType()
         {
-            return new TensorType(this);
+            return new TensorType(THSJIT_typeCast(handle));
         }
 
         internal DynamicType AsDynamicType()
         {
-            return new DynamicType(this);
+            return new DynamicType(THSJIT_typeCast(handle));
         }
 
         internal enum TypeKind : sbyte