TensorFlow.NET/src/TensorFlowNET.Core/Data/RangeDataset.cs at master · miltinhoc/TensorFlow.NET · GitHub

21 lines (19 loc) · 736 Bytes

using Tensorflow.Framework.Models;

using static Tensorflow.Binding;

namespace Tensorflow.Data

{

public class RangeDataset : DatasetSource

{

public RangeDataset(int stop,

int start = 0,

int step = 1,

TF_DataType output_type = TF_DataType.TF_INT64)

{

var start_tensor = tf.convert_to_tensor((long)start);

var step_tensor = tf.convert_to_tensor((long)step);

var stop_tensor = tf.convert_to_tensor((long)stop);

structure = new TensorSpec[] { new TensorSpec(new int[0], dtype: output_type) };

variant_tensor = ops.range_dataset(start_tensor, stop_tensor, step_tensor, output_types, output_shapes);

}

Expand file tree