code: def __init__(
      self,
      input=None,
      factor=None,
      axis=None,  # pylint: disable=redefined-builtin
      num_partitions=None,
      output_types=None,
      **kw,
  ):
      super().__init__(
          _input=input,
          _factor=factor,
          _axis=axis,
          _num_partitions=num_partitions,
          _output_types=output_types,
          **kw,
      )
  
imports: from ...None import opcodes
  from ...core import recursive_tile
  from ...core.context import get_context
  from ...serialization.serializables import KeyField, Float64Field, Int64Field
  from ...tensor.datasource import tensor
  from ...utils import has_unknown_shape, ceildiv
  from ..operands import TensorOperandMixin, TensorOperand
label_desc: Use image node_id 1 to create a new TensorRebalance object from inherited base classes: RebalanceMixin, TensorOperandMixin, TensorOperand with example: obj = TensorRebalance(input, factor, axis, num_partitions, output_types)