Pytorch和ONNX导出中的位移位运算符
我有一个来自框架的量化网络，我想将其导出为ONNX文件。量化需要添加执行按位右移的中间层以避免溢出。我必须在网络的其他层之间插入这些层 我想我可以使用Pytorch中的，以某种方式创建附加层。 我的第一个问题是：这样的层已经存在，还是我必须从头开始创建它们？例如，我可以使用带有对角权重矩阵的线性层，并告诉框架不要更改权重吗 然后，对于ONNX导出，我认为这样的层还不受支持。有没有一个简单的方法来进行导出 谢谢你的帮助Pytorch和ONNX导出中的位移位运算符,pytorch,bit-shift,quantization,Pytorch,Bit Shift,Quantization,我有一个来自框架的量化网络，我想将其导出为ONNX文件。量化需要添加执行按位右移的中间层以避免溢出。我必须在网络的其他层之间插入这些层 我想我可以使用Pytorch中的，以某种方式创建附加层。 我的第一个问题是：这样的层已经存在，还是我必须从头开始创建它们？例如，我可以使用带有对角权重矩阵的线性层，并告诉框架不要更改权重吗 然后，对于ONNX导出，我认为这样的层还不受支持。有没有一个简单的方法来进行导出 谢谢你的帮助