unflattened_size

Auto-generated from flax_nnx_code_defs

PyTorchApple MLXFlax NNX

PyTorch

API: torch.nn.modules.flatten.Unflatten.unflattened_size

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.core.unflatten

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: nnx.graph.PytreeNodeImpl.unflatten

Strategy: Direct Mapping

Official Docs ↗