sqrt

dragon.vm.torch.sqrt(
  input,
  out=None
)[source]

Compute the square root of input.

\[\text{out} = \sqrt{\text{input}} \]

Examples:

x = torch.tensor([4., 9., 16.])
print(torch.sqrt(x))  # [2., 3., 4.]
Parameters:
Returns:

dragon.vm.torch.Tensor – The output tensor.