示例#1
0
        public void WriteAndReadIntsViaMemoryFile()
        {
            var file = new Torch.IO.MemoryFile("rwb");

            Assert.IsNotNull(file);
            Assert.IsTrue(file.CanWrite);

            var data0 = new int[4];

            for (var i = 0; i < data0.Length; ++i)
            {
                data0[i] = (byte)i;
            }
            file.WriteInts(data0);
            Assert.AreEqual(data0.Length * sizeof(int), file.Position);
            file.Seek(0);

            var data1 = new int[data0.Length];
            var rd    = file.ReadInts(data1, data0.Length);

            Assert.AreEqual(rd, data0.Length);
            Assert.AreEqual(data0.Length * sizeof(int), file.Position);

            for (var i = 0; i < rd; ++i)
            {
                Assert.AreEqual(data0[i], data1[i]);
            }
            file.Close();
            Assert.IsFalse(file.IsOpen);
        }
示例#2
0
        public void WriteAndReadStorageIntsViaMemoryFile()
        {
            const int size = 10;

            var file = new Torch.IO.MemoryFile("rwb");

            Assert.IsNotNull(file);
            Assert.IsTrue(file.CanWrite);

            var storage0 = new TorchSharp.IntTensor.IntStorage(size);

            for (var i = 0; i < size; ++i)
            {
                storage0[i] = i;
            }

            file.WriteInts(storage0);
            Assert.AreEqual(size * sizeof(int), file.Position);
            file.Seek(0);

            var storage1 = new TorchSharp.IntTensor.IntStorage(size);
            var rd       = file.ReadInts(storage1);

            Assert.AreEqual(rd, size);
            Assert.AreEqual(size * sizeof(int), file.Position);

            for (var i = 0; i < rd; ++i)
            {
                Assert.AreEqual(storage0[i], storage1[i]);
            }

            file.Close();
            Assert.IsFalse(file.IsOpen);
        }