pyscfad.ml.gto.basis_array.make_ao_mask#

pyscfad.ml.gto.basis_array.make_ao_mask(basis, mask_shl, mask_ctr, cart=False)[source]#
Parameters:

  • basis (BasisArray)

  • mask_shl (jnp.Array)

  • mask_ctr (jnp.Array)

  • cart (bool)

Return type:

jax.Array