torch.zeros_like
torch.zeros_like(input, *, dtype=None, layout=None, device=None, requires_grad=False, memory_format=torch.preserve_format) -> Tensor
Returns a tensor filled with the scalar value 0, with the same
   size as "input". "torch.zeros_like(input)" is equivalent to
   "torch.zeros(input.size(), dtype=input.dtype, layout=input.layout,
   device=input.device)".
Warning:
 As of 0.4, this function does not support an "out" keyword. As an
 alternative, the old "torch.zeros_like(input, out=output)" is
 equivalent to "torch.zeros(input.size(), out=output)".

Parameters:
      input (Tensor) -- the size of "input" will determine size
      of the output tensor.
Keyword Arguments:
      * dtype ("torch.dtype", optional) -- the desired data type
        of returned Tensor. Default: if "None", defaults to the dtype
        of "input".
  * **layout** ("torch.layout", optional) -- the desired layout of
