DStackΒΆ

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

Abstract Signature:

DStack(tup: List[Tensor])

PyTorch

API: torch.dstack
Strategy: Direct Mapping

JAX (Core)

API: jax.numpy.dstack
Strategy: Direct Mapping

NumPy

API: numpy.dstack
Strategy: Direct Mapping

Keras

API: keras.ops.dstack
Strategy: Direct Mapping

TensorFlow

API: tf.experimental.numpy.dstack
Strategy: Direct Mapping

Apple MLX

API: β€”
Strategy: Custom / Partial