Rate this Page
★★★★★

torch.zeros#

torch.zeros(*size,*,out=None,dtype=None,layout=torch.strided,device=None,requires_grad=False)→Tensor#

Returns a tensor filled with the scalar value0, with the shape definedby the variable argumentsize.

Parameters

size (int...) – a sequence of integers defining the shape of the output tensor.Can be a variable number of arguments or a collection like a list or tuple.

Keyword Arguments

  • out (Tensor,optional) – the output tensor.

  • dtype (torch.dtype, optional) – the desired data type of returned tensor.Default: ifNone, uses a global default (seetorch.set_default_dtype()).

  • layout (torch.layout, optional) – the desired layout of returned Tensor.Default:torch.strided.

  • device (torch.device, optional) – the desired device of returned tensor.Default: ifNone, uses the current device for the default tensor type(seetorch.set_default_device()).device will be the CPUfor CPU tensor types and the current CUDA device for CUDA tensor types.

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

Example:

>>>torch.zeros(2,3)tensor([[ 0.,  0.,  0.],        [ 0.,  0.,  0.]])>>>torch.zeros(5)tensor([ 0.,  0.,  0.,  0.,  0.])