Jermmy / pytorch-quantization-demo

A simple network quantization demo using pytorch from scratch.
Apache License 2.0
497 stars 96 forks source link

请问这里的x.clone()是什么用意呢? #18

Closed crazy-JiangDongHua closed 1 year ago

crazy-JiangDongHua commented 1 year ago

https://github.com/Jermmy/pytorch-quantization-demo/blob/ac7298f2b5f9cbaa9125ff58c40645ce1f381101/module.py#L270

Jermmy commented 1 year ago

因为下一行 x[x < self.qi.zero_point] = self.qi.zero_point 是 inplace 的改动,为了不破坏输入数据,所以先 clone 一份。

crazy-JiangDongHua commented 1 year ago

原来如此,涨知识了,感谢博主