Shortcuts

Template Function torch::nn::parallel::replicate(const ModuleHolder<ModuleType>&, const std::vector<Device>&)

Function Documentation

template<typename ModuleType>
std::vector<ModuleHolder<ModuleType>> torch::nn::parallel::replicate(const ModuleHolder<ModuleType> &module, const std::vector<Device> &devices)

Replicates a module holder on the given list of devices.

This method allows calling replicate() with a module holder, such as Linear.

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