PyTorch
API: torch.adaptive_avg_pool1d
Strategy: Direct Mapping
JAX (Core)
API: jax.image.resize
Strategy: Macro 'jax.image.resize({input}, shape=(*{input}.shape[:-1], {output_size}), method='linear')'
Keras
API: keras.layers.GlobalAveragePooling1D
Strategy: Direct Mapping
TensorFlow
API: tf.keras.layers.GlobalAveragePooling1D
Strategy: Direct Mapping
Apple MLX
API: —
Strategy: Custom / Partial
Flax NNX
API: nnx.avg_pool
Strategy: Macro 'jax.image.resize({input}, shape=(*{input}.shape[:-1], {output_size}), method='linear')'