[pruning] Activation reconstruction (#60292)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60292
Added activation reconstruction in the `reconstruct` method
Test Plan:
`buck test mode/dev-nosan //caffe2/test:ao -- TestBasePruner`
https://pxl.cl/1KLl1
Reviewed By: z-a-f
Differential Revision: D29236569
fbshipit-source-id: 1ad085f4143eb9fa3efca51e00d810e0fdb7e9b1