Is there any way to count/profile all matrix multiplication within a model

autograd
1

Is there any way to count/profile all matrix multiplication within a model

2

Hi,

The profiler from the autograd will be able to give you this information (as long as you have autograd enabled).