Function broadcast_to

imperative/python/megengine/functional/tensor.py:543–561

r"""Broadcasts a tensor to given shape. Args: inp: input tensor. shape: target shape. Returns: output tensor. Examples: >>> import numpy as np >>> data = Tensor(np.arange(0, 3, dtype=np.float32).reshape(3)) >>> out = F.broadcast_to(data,

(inp: Tensor, shape: Union[int, Iterable[int]])

def broadcast_to(inp: Tensor, shape: Union[int, Iterable[int]]) -> Tensor:
    r"""Broadcasts a tensor to given shape.

    Args:
        inp: input tensor.
        shape: target shape.

    Returns:
        output tensor.

    Examples:
        >>> import numpy as np
        >>> data = Tensor(np.arange(0, 3, dtype=np.float32).reshape(3))
        >>> out = F.broadcast_to(data, (2, 3))
        >>> out.numpy()
        array([[0., 1., 2.],
               [0., 1., 2.]], dtype=float32)
    """
    return broadcast_cpp(inp, shape)


def concat(inps: Iterable[Tensor], axis: int = 0, device=None) -> Tensor: