TensorFlow.NET/src/TensorFlowNET.Core/Data/SkipDataset.cs at master · ehtick/TensorFlow.NET · GitHub

21 lines (19 loc) · 627 Bytes

using static Tensorflow.Binding;

namespace Tensorflow

{

/// <summary>

/// A `Dataset` skipping the first `count` elements from its input.

/// </summary>

public class SkipDataset : UnaryUnchangedStructureDataset

{

Tensor _count;

public SkipDataset(IDatasetV2 input_dataset,

int count) : base(input_dataset)

{

_count = tf.convert_to_tensor(count, dtype: dtypes.int64, name: "count");

variant_tensor = ops.skip_dataset(input_dataset.variant_tensor,

_count,

output_types, output_shapes);

}

Expand file tree