dimΒΆ

Auto-generated from flax_nnx_code_defs

PyTorch

API: torch.nn.modules.activation.LogSoftmax.dim
Strategy: Direct Mapping

Apple MLX

API: mlx.core.array.ndim
Strategy: Direct Mapping

Flax NNX

API: nnx.variablelib.AbstractVariable.ndim
Strategy: Direct Mapping