Is it possible to have multiple distinct return paths in a forward pass? I’ve seen in other posts whenever there is branching logic, they will return multiple values from the same return, kind of like
return value1, value2
Instead, I want to be able to have something like
… (potentially more layers)
Can the autograd engine support this? How would you set up your losses to know where you exited from and what layers to update? Apologies if this is documented somewhere but I haven’t been able to find it.