Exemplo n.º 1
0
        public void CreateHalfTensorOnes()
        {
            var shape = new long[] { 2, 2 };

            TorchTensor t = HalfTensor.Ones(shape);

            Assert.Equal(shape, t.Shape);
            Assert.Equal(1.0f, t.ReadHalf(0));
            Assert.Equal(1.0f, t.ReadHalf(3));
        }