Exemplo n.º 1
0
        public void TestSaveLoadTensorFloat()
        {
            var file = ".saveload.float.ts";

            if (File.Exists(file))
            {
                File.Delete(file);
            }
            var tensor = FloatTensor.Ones(new long[] { 5, 6 });

            tensor.Save(file);
            var tensorLoaded = TorchTensor.Load(file);

            File.Delete(file);
            Assert.NotNull(tensorLoaded);
            Assert.Equal(tensorLoaded.Type, tensor.Type);
            Assert.Equal(tensorLoaded, tensor);
        }