Merge pull request #16 from interesaaat/LibTorchSharpFirstTest

Added squeeze, index_select, and div by int

Author: interesaaat

TorchSharp/Tensor/TorchTensor.cs

@@ -293,6 +293,14 @@ public TorchTensor Grad()
             return new TorchTensor(THSTensor_grad(handle));
         }
 
+        [DllImport("libTorchSharp")]
+        extern static IntPtr THSTensor_index_select(IntPtr src, long dimension, IntPtr index);
+
+        public TorchTensor IndexSelect(long dimension, TorchTensor index)
+        {
+            return new TorchTensor(THSTensor_index_select(handle, dimension, index.Handle));
+        }
+
         [DllImport("libTorchSharp")]
         extern static IntPtr THSTensor_reshape(IntPtr src, IntPtr shape, int length);
 
@@ -307,6 +315,14 @@ public TorchTensor Reshape(params long[] shape)
             }
         }
 
+        [DllImport("libTorchSharp")]
+        extern static IntPtr THSTensor_squeeze(IntPtr src, long dimension);
+
+        public TorchTensor Squeeze(long dimension)
+        {
+            return new TorchTensor(THSTensor_squeeze(handle, dimension));
+        }
+
         [DllImport("libTorchSharp")]
         extern static IntPtr THSTensor_t(IntPtr src);
 
@@ -417,6 +433,14 @@ public void DivInPlace(TorchTensor target)
             THSTensor_div_(handle, target.Handle);
         }
 
+        [DllImport("libTorchSharp")]
+        extern static IntPtr THSTensor_divS(IntPtr src, int trg);
+
+        public TorchTensor Div(int target)
+        {
+            return new TorchTensor(THSTensor_divS(handle, target));
+        }
+
         [DllImport("libTorchSharp")]
         extern static IntPtr THSTensor_eq(IntPtr src, IntPtr trg);
 
@@ -573,6 +597,11 @@ public TorchTensor Sum(long[] dimensions, bool keepDimension = false)
             return left.Div(right);
         }
 
+        public static TorchTensor operator /(TorchTensor left, int right)
+        {
+            return left.Div(right);
+        }
+
         /// <summary>
         ///   Get a string representation of the tensor.
         /// </summary>