Merge pull request #10 from interesaaat/LibTorchSharpFirstTest

Author: interesaaat

TorchSharp/NN/Init.cs

@@ -0,0 +1,50 @@
+using System;
+using System.Runtime.InteropServices;
+using TorchSharp.Tensor;
+
+namespace TorchSharp.NN
+{
+    public static class Init
+    {
+        [DllImport("libTorchSharp")]
+        extern static void THSNN_initUniform(IntPtr src, double low, double high);
+
+        public static void Uniform<T>(ITorchTensor<T> tensor, double low = 0, double high = 1)
+        {
+            THSNN_initUniform(tensor.Handle, low, high);
+        }
+
+        [DllImport("libTorchSharp")]
+        extern static void THSNN_initKaimingUniform(IntPtr src, double a);
+
+        public static void KaimingUniform<T>(ITorchTensor<T> tensor, double a = 0)
+        {
+            THSNN_initKaimingUniform(tensor.Handle, a);
+        }
+
+        public static (long fanIn, long fanOut) CalculateFanInAndFanOut<T>(ITorchTensor<T> tensor)
+        {
+            var dimensions = tensor.Dimensions;
+
+            if (dimensions < 2)
+            {
+                throw new ArgumentException("Fan in and fan out can not be computed for tensor with fewer than 2 dimensions");
+            }
+
+            var shape = tensor.Shape;
+            // Linear
+            if (dimensions == 2)
+            {
+                return (shape[1], shape[2]);
+            }
+            else
+            {
+                var numInputFMaps = tensor.Shape[1];
+                var numOutputFMaps = tensor.Shape[0];
+                var receptiveFieldSize = tensor[0, 0].NumberOfElements;
+
+                return (numInputFMaps * receptiveFieldSize, numOutputFMaps * receptiveFieldSize);
+            }
+        }
+    }
+}

TorchSharp/NN/LossFunction.cs

@@ -34,9 +34,9 @@ public static ITorchTensor<float> NLL<T, U>(ITorchTensor<T> src, ITorchTensor<U>
         }
 
         [DllImport("libTorchSharp")]
-        extern static IntPtr THSNN_lossPoissonNLL(IntPtr srct, IntPtr trgt, bool logInput, bool full, double eps, long reduction);
+        extern static IntPtr THSNN_lossPoissonNLL(IntPtr srct, IntPtr trgt, bool logInput, bool full, float eps, long reduction);
 
-        public static ITorchTensor<float> PoissonNLL<T, U>(ITorchTensor<T> src, ITorchTensor<U> target, bool logInput = true, bool full = false, double eps = 1e-8, Reduction reduction = Reduction.Mean)
+        public static ITorchTensor<float> PoissonNLL<T, U>(ITorchTensor<T> src, ITorchTensor<U> target, bool logInput = true, bool full = false, float eps = 1e-8f, Reduction reduction = Reduction.Mean)
         {
             return new FloatTensor(THSNN_lossPoissonNLL(src.Handle, target.Handle, logInput, full, eps, (long)reduction));
         }

TorchSharp/Tensor/ITorchTensor.cs

@@ -16,7 +16,13 @@ public interface ITorchTensor<T> : IDisposable
 
         Span<T> Data { get; }
 
-        T Item { get; }
+        T DataItem { get; }
+
+        ITorchTensor<T> this[long i1] { get; }
+
+        ITorchTensor<T> this[long i1, long i2] { get; }
+
+        ITorchTensor<T> this[long i1, long i2, long i3] { get; }
 
         bool IsSparse { get; }