Replies: 1 comment
-
|
The argument was added here: #4299 It looks like the intended use-case was for numerical stability when summing a large array of small numbers. @zhangqiaorjc may be able to say more. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
From the documentation of
jax.ops.segment_sumone can specify abucket_sizeto improve numerical stability. However, there's little guidance on when to do this and what reasonable values are. So when and how should I pick this value?Thanks!
Beta Was this translation helpful? Give feedback.
All reactions