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

C++
1

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
2

bump. Anyone know the answer?

3

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