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.