Function squeeze

imperative/python/megengine/functional/tensor.py:1070–1087

r"""Removes dimension of shape 1. Args: inp: input tensor. axis: place of axis to be removed. Returns: output tensor. Examples: >>> import numpy as np >>> x = Tensor(np.array([1, 2], dtype=np.int32).reshape(1, 1, 2, 1)) >>> out = F.squee

(inp: Tensor, axis: Optional[Union[int, Sequence[int]]] = None)

def squeeze(inp: Tensor, axis: Optional[Union[int, Sequence[int]]] = None) -> Tensor:
    r"""Removes dimension of shape 1.

    Args:
        inp: input tensor.
        axis: place of axis to be removed.

    Returns:
        output tensor.

    Examples:
        >>> import numpy as np
        >>> x = Tensor(np.array([1, 2], dtype=np.int32).reshape(1, 1, 2, 1))
        >>> out = F.squeeze(x, 3)
        >>> out.numpy().shape
        (1, 1, 2)
    """
    return squeeze_cpp(inp, axis)


def repeat(inp: Tensor, repeats: int, axis: Optional[int] = None):