def METHOD_NAME(x: NNCFTensor, axis: int = 0) -> List[NNCFTensor]:
    tensor = x.tensor
    if list(tensor.shape) == []:  # pylint: disable=C1803
        tensor = tf.expand_dims(tensor, 0)
    tensor_list = tf.METHOD_NAME(tensor, axis=axis)
    return [TFNNCFTensor(t) for t in tensor_list]