public void GenerateSequenceLength() { var inputTensor = new Tensor(); var batchSize = 4; var generator = new SequenceLengthGenerator(); generator.Generate(inputTensor, batchSize, null); Assert.IsNotNull(inputTensor.Data as int[]); Assert.AreEqual((inputTensor.Data as int[])[0], 1); }
public void GenerateSequenceLength() { var inputTensor = new TensorProxy(); var alloc = new TensorCachingAllocator(); const int batchSize = 4; var generator = new SequenceLengthGenerator(alloc); generator.Generate(inputTensor, batchSize, null); Assert.IsNotNull(inputTensor.data); Assert.AreEqual(inputTensor.data[0], 1); alloc.Dispose(); }