forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathFilterDataset.cs
More file actions
58 lines (51 loc) · 2.04 KB
/
FilterDataset.cs
File metadata and controls
58 lines (51 loc) · 2.04 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
using System;
using Tensorflow.Functions;
using static Tensorflow.Binding;
namespace Tensorflow
{
/// <summary>
/// A `Dataset` that filters its input according to a predicate function.
/// </summary>
public class FilterDataset : UnaryDataset
{
public FilterDataset(IDatasetV2 input_dataset,
Func<Tensor, bool> predicate_func) : base(input_dataset)
{
Func<Tensors, Tensors> predicate_func_update = x =>
{
var result = predicate_func(x);
return constant_op.constant(result);
};
var func = new ConcreteFunction($"{predicate_func.Method.Name}_{Tensorflow.ops.uid_function()}");
func.Enter();
var inputs = new Tensors();
foreach (var input in input_dataset.element_spec)
inputs.Add(tf.placeholder(input.dtype, shape: input.shape, name: "arg"));
var outputs = predicate_func_update(inputs);
func.ToGraph(inputs, outputs);
func.Exit();
structure = func.OutputStructure;
variant_tensor = ops.filter_dataset(input_dataset.variant_tensor,
func,
output_types,
output_shapes);
}
public FilterDataset(IDatasetV2 input_dataset,
Func<Tensors, Tensors> predicate_func) : base(input_dataset)
{
var func = new ConcreteFunction($"{predicate_func.Method.Name}_{Tensorflow.ops.uid_function()}");
func.Enter();
var inputs = new Tensors();
foreach (var input in input_dataset.element_spec)
inputs.Add(tf.placeholder(input.dtype, shape: input.shape, name: "arg"));
var outputs = predicate_func(inputs);
func.ToGraph(inputs, outputs);
func.Exit();
structure = func.OutputStructure;
variant_tensor = ops.filter_dataset(input_dataset.variant_tensor,
func,
output_types,
output_shapes);
}
}
}