Function at::lstm_mps_backward¶

Function Documentation¶

inline ::std::tuple<at::Tensor, ::std::vector<at::Tensor>, ::std::vector<at::Tensor>> at::lstm_mps_backward(const ::std::optional<at::Tensor> &grad_y, const ::std::optional<at::Tensor> &grad_hy, const ::std::optional<at::Tensor> &grad_cy, const at::Tensor &z_state, const at::Tensor &cell_state_fwd, const at::Tensor &input, const at::Tensor &layersOutputs, at::TensorList hx, at::TensorList params, bool has_biases, int64_t num_layers, double dropout, bool train, bool bidirectional, bool batch_first)¶