Files
tensorneat-mend/examples/jax_playground.py
2023-05-11 19:04:56 +08:00

5 lines
82 B
Python

import jax.numpy as jnp
EMPTY_NODE = jnp.full((1, 5), jnp.nan)
print(EMPTY_NODE)