예제 #1
0
 public Tensor[] next()
 {
     try
     {
         return(ops.iterator_get_next(_iterator_resource, _dataset.output_types, _dataset.output_shapes));
     }
     catch (OutOfRangeError ex)
     {
         throw new StopIteration(ex.Message);
     }
 }
예제 #2
0
 public Tensor[] next()
 {
     try
     {
         var results = ops.iterator_get_next(_iterator_resource, _dataset.output_types, _dataset.output_shapes);
         foreach (var(i, tensor) in enumerate(results))
         {
             tensor.set_shape(_element_spec[i].shape);
         }
         return(results);
     }
     catch (OutOfRangeError ex)
     {
         throw new StopIteration(ex.Message);
     }
 }