Shortcuts

Template Function torch::nn::replicate_grad_edges

Function Documentation

template<typename ModuleType>
void torch::nn::replicate_grad_edges(const std::shared_ptr<Module> &module, const std::vector<std::shared_ptr<ModuleType>> &replicas, const std::vector<Device> &devices)

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