jax-js
    Preparing search index...

    Function split

    • Splits a PRNG key into num new keys by adding a leading axis.

      Parameters

      • key: Array
      • num: number | number[] = 2

      Returns Array