DStack

Stack arrays in sequence depth wise (along third axis).

Abstract Signature:

DStack(tup: List[Tensor])

PyTorch

API: torch.dstack
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.dstack
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.dstack
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.dstack
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.experimental.numpy.dstack
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: —
Strategy: Custom / Partial