brainpy.math.sqrt#

brainpy.math.sqrt = <function _compatible_with_brainpy_array.<locals>.new_fun>#

Similar to jax.numpy.sqrt function, while it is compatible with brainpy Array/Variable.

Note that this function is also compatible with:

  1. NumPy or PyTorch syntax when receiving out argument.

  2. PyTorch syntax when receiving keepdim or dim argument.

  3. TensorFlow syntax when receiving keep_dims argument.