Python np_or_jnp Beispiele

Programmiersprache: Python

Namespace / Paketname: gfsa.jax_util

Methode / Funktion: np_or_jnp

Beispiele auf hotexamples.com: 2

Python np_or_jnp - 2 Beispiele gefunden. Dies sind die am besten bewerteten Python Beispiele für die gfsa.jax_util.np_or_jnp, die aus Open Source-Projekten extrahiert wurden. Sie können Beispiele bewerten, um die Qualität der Beispiele zu verbessern.

Beispiel #1

Datei anzeigen

Datei: jax_util_test.py Projekt: MitchellTesla/google-research

    def test_np_or_jnp(self):
        self.assertIs(jax_util.np_or_jnp(1.), np)
        self.assertIs(jax_util.np_or_jnp(np.array(1.)), np)
        self.assertIs(jax_util.np_or_jnp(jnp.array(1.)), jnp)

        def trace_check(x):
            self.assertIs(jax_util.np_or_jnp(x), jnp)

        jax.make_jaxpr(trace_check)(np.array(1.))

Beispiel #2

Datei anzeigen

Datei: jax_util_test.py Projekt: MitchellTesla/google-research

 def trace_check(x):
     self.assertIs(jax_util.np_or_jnp(x), jnp)