class Tensorflow::Data::TensorSliceDataset

Public Class Methods

new(elements) click to toggle source
Calls superclass method Tensorflow::Data::Dataset::new
# File lib/tensorflow/data/tensor_slice_dataset.rb, line 4
def initialize(elements)
  @tensors = self.class.to_tensor_array(elements)
  @output_types = @tensors.map(&:dtype)
  @output_shapes = @tensors.map do |tensor|
    tensor.shape[1..]
  end
  variant_tensor = RawOps.tensor_slice_dataset(@tensors, toutput_types: @output_types, output_shapes: @output_shapes)
  super(variant_tensor)
end