@@ -14,6 +14,7 @@ namespace Tensorflow
1414 public class DatasetV2 : IDatasetV2
1515 {
1616 protected dataset_ops ops = new dataset_ops ( ) ;
17+ public string [ ] class_names { get ; set ; }
1718 public Tensor variant_tensor { get ; set ; }
1819
1920 public TensorSpec [ ] structure { get ; set ; }
@@ -54,7 +55,7 @@ public IDatasetV2 skip(int count)
5455 public IDatasetV2 optimize ( string [ ] optimizations , string [ ] optimization_configs )
5556 => new OptimizeDataset ( this , optimizations , optimization_configs : optimization_configs ) ;
5657
57- public IDatasetV2 map ( Func < Tensor , Tensor > map_func ,
58+ public IDatasetV2 map ( Func < Tensors , Tensors > map_func ,
5859 bool use_inter_op_parallelism = true ,
5960 bool preserve_cardinality = true ,
6061 bool use_legacy_function = false )
@@ -64,7 +65,7 @@ public IDatasetV2 map(Func<Tensor, Tensor> map_func,
6465 preserve_cardinality : preserve_cardinality ,
6566 use_legacy_function : use_legacy_function ) ;
6667
67- public IDatasetV2 map ( Func < Tensors , Tensors > map_func , int num_parallel_calls = - 1 )
68+ public IDatasetV2 map ( Func < Tensors , Tensors > map_func , int num_parallel_calls )
6869 => new ParallelMapDataset ( this , map_func , num_parallel_calls : num_parallel_calls ) ;
6970
7071 public IDatasetV2 flat_map ( Func < Tensor , IDatasetV2 > map_func )
0 commit comments