torch.full_like
函数定义
torch.full_like(input, fill_value, *, dtype=None,
layout=torch.strided, device=None, requires_grad=False,
memory_format=torch.preserve_format) → Tensor
Returns a tensor with the same size as input filled with fill_value. torch.full_like(input, fill_value)
is equivalent to torch.full(input.size(), fill_value, dtype=input.dtype, layout=input.layout, device=input.device)
.
参数
- input (Tensor) – the size of input will determine size of the output tensor.
- fill_value – the number to fill the output tensor with.
关键字参数
- 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 returned tensor. Default: if None, defaults to the layout of input.
- device (torch.device, optional) – the desired device of returned tensor. Default: if None, defaults to the device of input.
- requires_grad (bool, optional) – If autograd should record operations on the returned tensor. Default: False.
- memory_format (torch.memory_format, optional) – the desired memory format of returned Tensor. Default: torch.preserve_format.
例子
>>> torch.full((2, 3), 3.141592)
tensor([[ 3.1416, 3.1416, 3.1416],
[ 3.1416, 3.1416, 3.1416]])