jax-ml/jax
lax.conv_transpose has different convention than TF, document it and make switching easy?
オープン
#14,337 opened on 2023/02/07
documentationenhancementgood first issue
Repository metrics
- Stars
- (36,180 個のスター)
- PR merge metrics
- (平均マージ 2d 19h) (30d で 439 merged PRs)
説明
See screenshot:
Though the example uses Haiku, Haiku is just calling through to lax.conv_transpose.
The difference is in reversing the spatial axes of the kernel.
We can do at least two things to make this less surprising and more easily fixable:
- document the difference, and
- add an argument to
lax.conv_transposeto instead use the TF convention.