Export all operands from each conv layer

Hello everyone,

I’m interested in exporting each operation from a convolutional layer. Is there a straightforward method to accomplish this? The goal is to create a table with columns for op1 and op2. These operands will be used to test the circuits I’ve constructed.

I’m aware of the approach involving adjustments to the model class and the development of routines to export everything. However, is there a simpler method, perhaps utilizing pre-existing PyTorch functionalities, such as monitoring and storing or something similar?