forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathIDatasetV2.cs
More file actions
101 lines (75 loc) · 3.14 KB
/
IDatasetV2.cs
File metadata and controls
101 lines (75 loc) · 3.14 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
using System;
using System.Collections.Generic;
using Tensorflow.Framework.Models;
namespace Tensorflow
{
public interface IDatasetV2 : IEnumerable<(Tensors, Tensors)>
{
string[] class_names { get; set; }
Tensor variant_tensor { get; set; }
Shape[] output_shapes { get; }
TF_DataType[] output_types { get; }
TensorSpec[] element_spec { get; }
TensorSpec[] structure { get; set; }
int FirstInputTensorCount { get; set; }
/// <summary>
/// Caches the elements in this dataset.
/// </summary>
/// <param name="filename"></param>
/// <returns></returns>
IDatasetV2 cache(string filename = "");
/// <summary>
/// Creates a `Dataset` by concatenating the given dataset with this dataset.
/// </summary>
/// <param name="dataset"></param>
/// <returns></returns>
IDatasetV2 concatenate(IDatasetV2 dataset);
/// <summary>
///
/// </summary>
/// <param name="count"></param>
/// <returns></returns>
IDatasetV2 repeat(int count = -1);
/// <summary>
/// Creates a `Dataset` that includes only 1/`num_shards` of this dataset.
/// </summary>
/// <param name="num_shards">The number of shards operating in parallel</param>
/// <param name="index">The worker index</param>
/// <returns></returns>
IDatasetV2 shard(int num_shards, int index);
IDatasetV2 shuffle(int buffer_size, int? seed = null, bool reshuffle_each_iteration = true);
/// <summary>
/// Creates a `Dataset` that skips `count` elements from this dataset.
/// </summary>
/// <param name="count"></param>
/// <returns></returns>
IDatasetV2 skip(int count);
IDatasetV2 batch(int batch_size, bool drop_remainder = false);
IDatasetV2 prefetch(int buffer_size = -1, int? slack_period = null);
IDatasetV2 take(int count);
IDatasetV2 optimize(string[] optimizations, string[] optimization_configs);
IDatasetV2 map(Func<Tensors, Tensors> map_func,
bool use_inter_op_parallelism = true,
bool preserve_cardinality = true,
bool use_legacy_function = false);
IDatasetV2 map(Func<Tensors, Tensors> map_func,
int num_parallel_calls);
IDatasetV2 filter(Func<Tensors, Tensors> map_func);
IDatasetV2 filter(Func<Tensor, bool> map_func);
OwnedIterator make_one_shot_iterator();
IDatasetV2 flat_map(Func<Tensor, IDatasetV2> map_func);
IDatasetV2 model(AutotuneAlgorithm algorithm, long cpu_budget, long ram_budget);
IDatasetV2 with_options(DatasetOptions options);
/// <summary>
/// Apply options, such as optimization configuration, to the dataset.
/// </summary>
/// <returns></returns>
IDatasetV2 apply_options();
/// <summary>
/// Returns the cardinality of `dataset`, if known.
/// </summary>
/// <param name="name"></param>
/// <returns></returns>
Tensor cardinality(string name = null);
}
}