Merge pull request #27 from interesaaat/LibTorchSharpFirstTest

Add eq and add with scalars

Author: interesaaat

TorchSharp/Tensor/TorchTensor.cs

@@ -403,6 +403,14 @@ public TorchTensor Add(TorchTensor target, int scalar = 1)
             return new TorchTensor(THSTensor_add(handle, scalar, target.Handle));
         }
 
+        [DllImport("libTorchSharp")]
+        extern static IntPtr THSTensor_addS(IntPtr src, IntPtr trg);
+
+        public TorchTensor Add(Scalar scalar)
+        {
+            return new TorchTensor(THSTensor_addS(handle, scalar.Handle));
+        }
+
         [DllImport("libTorchSharp")]
         extern static void THSTensor_add_(IntPtr src, int scalar, IntPtr trg);
 
@@ -483,6 +491,14 @@ public TorchTensor Eq(TorchTensor target)
             return new TorchTensor(THSTensor_eq(handle, target.Handle));
         }
 
+        [DllImport("libTorchSharp")]
+        extern static IntPtr THSTensor_eqS(IntPtr src, IntPtr trg);
+
+        public TorchTensor Eq(Scalar target)
+        {
+            return new TorchTensor(THSTensor_eq(handle, target.Handle));
+        }
+
         [DllImport("libTorchSharp")]
         extern static bool THSTensor_equal(IntPtr src, IntPtr trg);
 
@@ -616,6 +632,11 @@ public TorchTensor Sum(long[] dimensions, bool keepDimension = false)
             return left.Add(right);
         }
 
+        public static TorchTensor operator +(TorchTensor left, Scalar right)
+        {
+            return left.Add(right);
+        }
+
         public static TorchTensor operator *(TorchTensor left, TorchTensor right)
         {
             return left.Mul(right);