lib/tensor_stream/ops.rb
def dynamic_partition(data, partitions, num_partitions, name: nil)
      result = _op(:dynamic_partition, data, partitions, num_partitions: num_partitions, name: nil)
      num_partitions.times.map do |index|
        result[index]
      end
    end
def dynamic_partition(data, partitions, num_partitions, name: nil)
      result = _op(:dynamic_partition, data, partitions, num_partitions: num_partitions, name: nil)
      num_partitions.times.map do |index|
        result[index]
      end
    end
Partitions data into num_partitions tensors using indices from partitions