Setting LSTM Parameters

C++
1

I have 4 torch::Tensor in C++ that I want to set as the {weight,bias}_{i,h}h parameters on a LSTM… what is the correct method to do this?

2

Something like?

 auto lstm_params = lstm->named_parameters();

 lstm_params.find("weight_ih_l0")->data() = w_ih;
 lstm_params.find("weight_hh_l0")->data() = w_hh;
 lstm_params.find("bias_ih_l0")->data() = b_ih;
 lstm_params.find("bias_hh_l0")->data() = b_hh;