TensorFlow.NET/src/TensorFlowNET.Core/Numpy/NDArray.Creation.cs

using System;
using System.Collections.Generic;
using System.Text;
using Tensorflow.Eager;
using static Tensorflow.Binding;

namespace Tensorflow.NumPy
{
    public partial class NDArray
    {
        public NDArray(bool value) => Init(value);
        public NDArray(byte value) => Init(value);
        public NDArray(short value) => Init(value);
        public NDArray(int value) => Init(value);
        public NDArray(long value) => Init(value);
        public NDArray(float value) => Init(value);
        public NDArray(double value) => Init(value);
        public NDArray(Array value, Shape? shape = null) => Init(value, shape);
        public NDArray(Shape shape, TF_DataType dtype = TF_DataType.TF_DOUBLE) => Init(shape, dtype: dtype);
        public NDArray(Tensor value, Shape? shape = null) => Init(value, shape);
        public NDArray(byte[] bytes, TF_DataType dtype) => Init(bytes, dtype);

        public static NDArray Scalar<T>(T value) where T : unmanaged
            => value switch
            {
                bool val => new NDArray(val),
                byte val => new NDArray(val),
                int val => new NDArray(val),
                long val => new NDArray(val),
                float val => new NDArray(val),
                double val => new NDArray(val),
                _ => throw new NotImplementedException("")
            };

        void Init<T>(T value) where T : unmanaged
        {
            _tensor = value switch
            {
                bool val => new Tensor(val),
                byte val => new Tensor(val),
                int val => new Tensor(val),
                long val => new Tensor(val),
                float val => new Tensor(val),
                double val => new Tensor(val),
                _ => throw new NotImplementedException("")
            };

            _tensor.SetReferencedByNDArray();
        }

        void Init(Array value, Shape? shape = null)
        {
            _tensor = new Tensor(value, shape ?? value.GetShape());
            _tensor.SetReferencedByNDArray();
        }

        void Init(Shape shape, TF_DataType dtype = TF_DataType.TF_DOUBLE)
        {
            _tensor = new Tensor(shape, dtype: dtype);
            _tensor.SetReferencedByNDArray();
        }

        void Init(Tensor value, Shape? shape = null)
        {
            // created tensor in graph mode
            if (value.TensorDataPointer == IntPtr.Zero)
                value = tf.defaultSession.eval(value);

            _tensor = new Tensor(value.TensorDataPointer, shape ?? value.shape, value.dtype);
            _tensor.SetReferencedByNDArray();
        }

        void Init(byte[] bytes, TF_DataType dtype)
        {

        }
    }
}