ones_like¶

dragon.vm.torch.ones_like(
  input,
  dtype='float32',
  device=None,
  requires_grad=False
)[source]¶

Return a tensor of ones with shape as the other.

$$\text{out} \leftarrow 1$$

Parameters:

• input (dragon.vm.torch.Tensor) – The tensor for indicating the output shape.
• dtype (str, optional, default='float32') – 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.