remove_axis

Auto-generated from flax_nnx_code_defs

NumPy

API: numpy.nditer.remove_axis
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: nnx.bridge.variables.NNXMeta.remove_axis
Strategy: Direct Mapping
Official Docs ↗