TensorBenchmark.cs

using BenchmarkDotNet.Attributes;
using Tensorflow;
using Tensorflow.Eager;

namespace TensorFlowBenchmark
{
    [SimpleJob(launchCount: 1, warmupCount: 1, targetCount: 10)]
    [MinColumn, MaxColumn, MeanColumn, MedianColumn]
    public class TensorBenchmark
    {
        private double[] data;

        [GlobalSetup]
        public void Setup()
        {
            data = new double[100];
        }

        /*[Benchmark]
        public void ScalarTensor()
        {
            var g = new Graph();
            for (int i = 0; i < 100; i++)
            {
                using (var tensor = new Tensor(17.0))
                {

                }
            }
        }

        [Benchmark]
        public unsafe void TensorFromFixedPtr()
        {
            var g = new Graph();
            for (int i = 0; i < 100; i++)
            {
                fixed (double* ptr = &data[0])
                {
                    using (var t = new Tensor((IntPtr)ptr, new long[] { data.Length }, tf.float64, 8 * data.Length))
                    {
                    }
                }
            }
        }

        [Benchmark]
        public void TensorFromArray()
        {
            var g=new Graph();
            for (int i = 0; i < 100; i++)
            {
                using (var tensor = new Tensor(data))
                {

                }
            }
        }


        [Benchmark]
        public void TensorFromNDArray()
        {
            var g = new Graph();
            for (int i = 0; i < 100; i++)
            {
                using (var tensor = new Tensor(new NDArray(data)))
                {

                }
            }
        }*/

        [Benchmark]
        public void RawTensorV1()
        {
            var c = new EagerTensor(new float[,]
            {
                { 3.0f, 1.0f },
                { 1.0f, 2.0f }
            }, "");
        }

        [Benchmark]
        public void RawTensorV2()
        {
            var c = new EagerTensorV2(new float[,]
            {
                { 3.0f, 1.0f },
                { 1.0f, 2.0f }
            });
        }
    }
}