pytaco.tensor_sqrt
- pytaco.tensor_sqrt(t1, out_format, dtype=None)
Takes the square root of each input in the tensor.
Values must be >= 0.
- Parameters
- t1: tensor, array_like
input tensor or array_like object
- out_format: format, mode_format, optional
If a
format
is specified, the result tensor is stored in the format out_format.If a
mode_format
is specified, the result the result tensor has a with all of the dimensions stored in themode_format
passed in.
- dtype: Datatype
The datatype of the output tensor.
- Returns
- sqrt: tensor
The element wise sqrt of the input tensor.
Examples
>>> import pytaco as pt >>> pt.tensor_sqrt([4, 16, 9], out_format=pt.dense, dtype=pt.float32).toarray() array([2., 4., 3.], dtype=float32)