Function zeros

imperative/python/megengine/functional/tensor.py:359–389

r"""Returns a new tensor having a specified shape and filled with zeros. Args: shape(int...): the shape of the output tensor. Keyword args: dtype(:attr:`.Tensor.dtype`, optional): output tensor data type. device(:attr:`.Tensor.device`, optional): device on which to

(
    shape: Union[int, Tuple[int, ...]],
    *,
    dtype="float32",
    device: Optional[CompNode] = None
)

def zeros(
    shape: Union[int, Tuple[int, ...]],
    *,
    dtype="float32",
    device: Optional[CompNode] = None
) -> Tensor:
    r"""Returns a new tensor having a specified shape and filled with zeros.

    Args:
        shape(int...): the shape of the output tensor.

    Keyword args:
        dtype(:attr:`.Tensor.dtype`, optional): output tensor data type.
        device(:attr:`.Tensor.device`, optional): device on which to place the created tensor.

    Returns:
        a tensor containing zeros.

    Examples:
        >>> F.zeros((2, 3))
        Tensor([[0. 0. 0.]
         [0. 0. 0.]], device=xpux:0)
    """
    if isinstance(shape, int):
        shape = (shape,)
    if device == None:
        device = get_default_device()
    op = builtin.Fill(0, dtype)
    shape = astensor1d(shape, dtype="int32", device=device)
    (x,) = apply(op, shape)
    return x


def zeros_like(inp: Tensor) -> Tensor: