forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathZipDataset.cs
More file actions
22 lines (21 loc) · 697 Bytes
/
ZipDataset.cs
File metadata and controls
22 lines (21 loc) · 697 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
using System.Collections.Generic;
using System.Linq;
using Tensorflow.Framework.Models;
namespace Tensorflow
{
public class ZipDataset : DatasetV2
{
// keep all dataset references
IDatasetV2[] _inputs;
public ZipDataset(params IDatasetV2[] ds)
{
_inputs = ds;
var input_datasets = ds.Select(x => x.variant_tensor).ToArray();
var _structure = new List<TensorSpec>();
foreach (var dataset in ds)
_structure.AddRange(dataset.structure);
structure = _structure.ToArray();
variant_tensor = ops.zip_dataset(input_datasets, output_types, output_shapes);
}
}
}