Function at::gru_cell

• Defined in File Functions.h

Function Documentation

inline at::Tensor at::gru_cell(const at::Tensor &input, const at::Tensor &hx, const at::Tensor &w_ih, const at::Tensor &w_hh, const ::std::optional<at::Tensor> &b_ih = {}, const ::std::optional<at::Tensor> &b_hh = {})