``brainunit.lax`` module ======================== .. currentmodule:: brainunit.lax .. automodule:: brainunit.lax Functions that Accepting Unitless --------------------------------- .. autosummary:: :toctree: generated/ :nosignatures: :template: classtemplate.rst acos acosh asin asinh atan atanh collapse cumlogsumexp bessel_i0e bessel_i1e digamma lgamma erf erfc erf_inv logistic atan2 polygamma igamma igammac igamma_grad_a random_gamma_grad zeta betainc shift_left shift_right_arithmetic shift_right_logical fft collapse Array Creation Functions ------------------------ .. autosummary:: :toctree: generated/ :nosignatures: :template: classtemplate.rst zeros_like_array iota broadcasted_iota Functions that Changing Unit ---------------------------- .. autosummary:: :toctree: generated/ :nosignatures: :template: classtemplate.rst rsqrt div pow integer_pow mul rem batch_matmul conv conv_transpose dot_general Functions that Keeping Unit --------------------------- .. autosummary:: :toctree: generated/ :nosignatures: :template: classtemplate.rst slice dynamic_slice dynamic_update_slice gather index_take slice_in_dim index_in_dim dynamic_slice_ind_dim dynamic_index_in_dim dynamic_update_slice_in_dim dynamic_update_index_in_dim sort sort_key_val neg cummax cummin cumsum scatter scatter_add scatter_sub scatter_mul scatter_min scatter_max scatter_apply sub complex pad clamp convert_element_type bitcast_convert_type approx_max_k approx_min_k top_k broadcast broadcast_in_dim broadcast_to_rank Functions that Removing Unit ---------------------------- .. autosummary:: :toctree: generated/ :nosignatures: :template: classtemplate.rst population_count clz eq ne ge gt le lt Linalg Functions ---------------- .. autosummary:: :toctree: generated/ :nosignatures: :template: classtemplate.rst cholesky eig eigh hessenberg lu qdwh qr schur svd tridiagonal householder_product triangular_solve tridiagonal_solve Other Functions --------------- .. autosummary:: :toctree: generated/ :nosignatures: :template: classtemplate.rst reduce reduce_precision broadcast_shapes