torch.profiler.KinetoStep.key_averages
KinetoStep.key_averages(_options?: KeyAveragesOptions): KinetoStepGroups events by input shape for more detailed analysis.
Parameters
_optionsKeyAveragesOptionsoptional
Returns
KinetoStep– This KinetoStep for method chainingExamples
const profiler = new Profiler();
profiler.start();
// ... run operations ...
await profiler.stop();
const stats = profiler.key_averages({ group_by_input_shape: true });