Can data_parallel take multiple input tensors and can it be used during backward() (libtorch)

Can torch::nn::parallel::data_parallel support a Module that takes multiple input tensors for its forward() ?

Also does it work for backwards()?

1 Like

bump. Anyone know the answer?

Based on this code it seems that a single Tensor is required.