1.torch.jit.script将pt转成jit
2.libtorch
前向传播若有多个参数,必须将参数打包成 torch::jit::IValue
的列表
std::vector<torch::jit::IValue> inputs;
inputs.push_back(obs);
inputs.push_back(z);
torch::Tensor action = this->actor.forward(inputs).toTensor();
1.torch.jit.script将pt转成jit
2.libtorch
前向传播若有多个参数,必须将参数打包成 torch::jit::IValue
的列表
std::vector<torch::jit::IValue> inputs;
inputs.push_back(obs);
inputs.push_back(z);
torch::Tensor action = this->actor.forward(inputs).toTensor();