brainpy.math.amax#
- brainpy.math.amax(a, axis=None, out=None, keepdims=<no value>, initial=<no value>, where=<no value>)#
Similar to
jax.numpy.amax
function, while it is compatible with brainpy Array/Variable.Note that this function is also compatible with:
NumPy or PyTorch syntax when receiving
out
argument.PyTorch syntax when receiving
keepdim
ordim
argument.TensorFlow syntax when receiving
keep_dims
argument.