Array API Standard Support: spatial.transform#
Support on CPU#
Legend
✔️ = supported
✖ = unsupported
N/A = out-of-scope
blank = not currently documented
function/class |
torch |
jax |
dask |
|---|---|---|---|
✔️ |
✔️ |
✖ |
|
✔️ |
✔️ |
✖ |
|
✔️ |
✔️ |
✖ |
Support on GPU#
Legend
✔️ = supported
✖ = unsupported
N/A = out-of-scope
blank = not currently documented
function/class |
cupy |
torch |
jax |
|---|---|---|---|
✔️ |
✔️ |
✔️ |
|
✔️ |
✔️ |
✔️ |
|
✔️ |
✔️ |
✔️ |
Support with JIT#
Legend
✔️ = supported
✖ = unsupported
N/A = out-of-scope
blank = not currently documented
function/class |
jax |
|---|---|
✔️ |
|
✔️ |
|
✔️ |