Tabulate

Generates a summary table of the model architecture.

Abstract Signature:

Tabulate(model, input_size)

PyTorch

API: torchinfo.summary
Strategy: Direct Mapping
Official Docs ↗

Keras

API: summary
Strategy: Plugin (method_call)

Flax NNX

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