Jax: Document `lax.stop_gradient`

Created on 10 Mar 2019  路  1Comment  路  Source: google/jax

Is there a function similar to tensorflow's tf.stop_gradient?

How can I do this in jax??

documentation
Source
picture act65

Most helpful comment

It's not documented, but I think you're looking for jax.lax.stop_gradient.

We should document it!

picture hawkinsp on 10 Mar 2019
👍3

>All comments

It's not documented, but I think you're looking for jax.lax.stop_gradient.

We should document it!

hawkinsp picture hawkinsp on 10 Mar 2019
👍3
Was this page helpful?
0 / 5 - 0 ratings

Related issues

sursu picture sursu  路  3Comments
alexbw picture alexbw  路  3Comments
lonelykid picture lonelykid  路  3Comments
yfji picture yfji  路  3Comments
madvn picture madvn  路  3Comments