Conv3d

Applies a 3D convolution.

Abstract Signature:

Conv3d(input: Tensor, weight: Tensor, bias: Tensor, stride: int = 1, padding: int = 0, dilation: int = 1, groups: int = 1)

PyTorch

API: torch.nn.functional.conv3d
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.lax.conv_general_dilated
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.layers.Conv3D
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.keras.layers.Conv3D
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.conv3d
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.Conv
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: praxis.layers.Conv3D
Strategy: Direct Mapping
Official Docs ↗