torch.onnx.export(）转换onnx文件时，怎么将权重参数张量转移到cpu？

在torch.onnx.export()函数中，可以通过指定参数"export_params=True"来将权重参数张量也导出到ONNX文件中。同时，可以通过在转换前将所有张量移动到CPU上，即使用.cpu()函数来实现将权重参数张量转移到CPU上。具体示例如下：

import torch

# 构建模型
class MyModel(torch.nn.Module):
def __init__(self):
super(MyModel, self).__init__()
self.linear1 = torch.nn.Linear(10, 10)
self.linear2 = torch.nn.Linear(10, 10)

def forward(self, x):
x = self.linear1(x)
x = self.linear2(x)
return x

model = MyModel()

# 生成随机输入
x = torch.rand(1, 10)

# 将模型所有张量移动到CPU
model.cpu()

# 导出ONNX文件
torch.onnx.export(model, x, "mymodel.onnx", export_params=True)