C# (CSharp) BatchSizeGenerator Examples

Example #1

        public void GenerateBatchSize()
        {
            var inputTensor = new Tensor();
            var batchSize   = 4;
            var generator   = new BatchSizeGenerator();

            generator.Generate(inputTensor, batchSize, null);
            Assert.IsNotNull(inputTensor.Data as int[]);
            Assert.AreEqual((inputTensor.Data as int[])[0], batchSize);
        }

Example #2

File: EditModeTestInternalBrainTensorGenerator.cs Project: needle-mirror/com.unity.ml-agents

        public void GenerateBatchSize()
        {
            var       inputTensor = new TensorProxy();
            var       alloc       = new TensorCachingAllocator();
            const int batchSize   = 4;
            var       generator   = new BatchSizeGenerator(alloc);

            generator.Generate(inputTensor, batchSize, null);
            Assert.IsNotNull(inputTensor.data);
            Assert.AreEqual(inputTensor.data[0], batchSize);
            alloc.Dispose();
        }