def __init__(self, axis: int = 2, n: Union[int, None] = None) -> None:
if PT_BEFORE_1_7:
raise InvalidPyTorchVersionError("1.7.0", self.__class__.__name__)
super().__init__()
self.axis = axis
self.n = n
def __init__(self, axis: int = 1, n: Union[int, None] = None) -> None:
if PT_BEFORE_1_7:
raise InvalidPyTorchVersionError("1.7.0", self.__class__.__name__)
if axis < 0:
raise ValueError("axis must be zero or positive.")
self.axis = axis
self.n = n