Open
Description
Currently monai.data.TestTimeAugmentation
seemingly assumes the inverse transformations can be applied to the predictions, which necessarily means the predictions must be spatial.
Seems fairly straightforward - would be willing to write up a PR if there's interest and no reason this doesn't make sense.