sqrt

nvtripy.sqrt(input: Tensor) Tensor[source]

Computes the elementwise square root of the elements of the input tensor.

Parameters:

input (Tensor) – [dtype=T1] The input tensor.

Returns:

[dtype=T1] A new tensor of the same shape as the input tensor.

Return type:

Tensor

DATA TYPE CONSTRAINTS:
Example
1input = tp.arange(3, dtype=tp.float32) + 1.0
2output = tp.sqrt(input)
Local Variables
>>> input
tensor([1, 2, 3], dtype=float32, loc=gpu:0, shape=(3,))

>>> output
tensor([1, 1.41421, 1.73205], dtype=float32, loc=gpu:0, shape=(3,))