Suggestion to modify paper

[NoahYn]

I stop to thinking about Figure 4.
you discribe 3x3 conv params as four 3x3 matrices, but 1x1 conv params as a 2x2 matrix not four 1x1 matrices. This is trivial but confusing for me. I think I can understand more quickly if it is represented four 1x1 matrices.
Also I want to know which I understand the Figure 4 correctly or not.

Anyway, this is very impressive work! I wanted to fasten the inference time and I found your work! thank you.