You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
It looks like the jax.numpy version of arange doesn't take symbolic inputs (like reshape in #43). Currently, the test for that Op is marked as an expected failure.
Perhaps we can put together an implementation using jax.lax?
The text was updated successfully, but these errors were encountered:
jax.numpy.arange and jax.numpy.reshape only accept concrete values respectively for their start, stop, step and shape parameters. Concrete values are either constants or the output of a Shape operator or combination of such outputs with Python operator. This is being refactored in #1338.
It looks like the
jax.numpy
version ofarange
doesn't take symbolic inputs (likereshape
in #43). Currently, the test for thatOp
is marked as an expected failure.Perhaps we can put together an implementation using
jax.lax
?The text was updated successfully, but these errors were encountered: