Shortcuts

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 at::Tensor &grad_y, const c10::optional<at::Tensor> &grad_hy, const c10::optional<at::Tensor> &grad_cy, const at::Tensor &z_state, const at::Tensor &cell_state_fwd, const at::Tensor &input, at::TensorList hx, at::TensorList params, bool has_biases, int64_t num_layers, double dropout, bool train, bool bidirectional, bool batch_first)

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources