Permute the axes/dimensions of an array.
>>> a = jnp.array([[1, 2, 3],
... [4, 5, 6]])
>>> jnp.permute_dims(a, (1, 0))
Array([[1, 4],
[2, 5],
[3, 6]], dtype=int32)
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.3