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