This is purely a typing issue

Many functions in tianshou deal with union types, e.g. an input can be a Batch, a Tensor or ndarray and the output types depend on the input. Python's @overload can provide a stricter type annotation in these cases.

The description of the issue is currently incomplete and will be extended as it is addressed