sakura.utils.size_estimator.SizeEstimator.get_output_sizes

SizeEstimator.get_output_sizes()

Determine output dimensions for each layer by running a sample input through the model.