﻿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);
        }
    }
}
