Rate this Page
★★★★★

torch.full#

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

Creates a tensor of sizesize filled withfill_value. Thetensor’s dtype is inferred fromfill_value.

Parameters:

  • size (int...) – a list, tuple, ortorch.Size of integers defining theshape of the output tensor.

  • fill_value (Scalar) – the value to fill the output tensor with.

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.full((2,3),3.141592)tensor([[ 3.1416,  3.1416,  3.1416],        [ 3.1416,  3.1416,  3.1416]])