Multiple outputs implementation

[Soumya-dutta]

Hello,

Suppose my model returns multiple outputs. How should the functional approach be modified to handle this?

Thanks.

[luyug]

With the grad cache decorator, you can return a tuple of tensors.