full_like¶
dragon.vm.torch.
full_like
(
input,
fill_value,
out=None,
dtype='int64',
device=None,
requires_grad=False
)[source]¶Return a tensor filled with a scalar with size as input.
Examples:
print(torch.full_like(torch.zeros(1, 2), 1)) # [[1, 1]]
- Parameters:
- input (dragon.vm.torch.Tensor) – The tensor for indicating shape.
- fill_value (number) – The scalar to fill.
- out (dragon.vm.torch.Tensor, optional) – The output tensor.
- dtype (str, optional, default='int64') – The data type of output tensor.
- device (dragon.vm.torch.device, optional) – The device of output tensor.
- requires_grad (bool, optional, default=False) – Record gradient for output tensor or not.
- Returns:
dragon.vm.torch.Tensor – The output tensor.