Rate this Page
★★★★★

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 value0, with the same size asinput.torch.zeros_like(input) is equivalent totorch.zeros(input.size(),dtype=input.dtype,layout=input.layout,device=input.device).

Warning

As of 0.4, this function does not support anout keyword. As an alternative,the oldtorch.zeros_like(input,out=output) is equivalent totorch.zeros(input.size(),out=output).

Parameters

input (Tensor) – the size ofinput will determine size of the output tensor.

Keyword Arguments

  • dtype (torch.dtype, optional) – the desired data type of returned Tensor.Default: ifNone, defaults to the dtype ofinput.

  • layout (torch.layout, optional) – the desired layout of returned tensor.Default: ifNone, defaults to the layout ofinput.

  • device (torch.device, optional) – the desired device of returned tensor.Default: ifNone, defaults to the device ofinput.

  • requires_grad (bool,optional) – If autograd should record operations on thereturned tensor. Default:False.

  • memory_format (torch.memory_format, optional) – the desired memory format ofreturned Tensor. Default:torch.preserve_format.

Example:

>>>input=torch.empty(2,3)>>>torch.zeros_like(input)tensor([[ 0.,  0.,  0.],        [ 0.,  0.,  0.]])