How do i get the size of preds and truth tensors after i call .update()

[P-Sood]

I am running my model using DDP, and 4 GPUS, and I want to make sure that the metrics being computed are consistent by checking their size after and before a reset, as well as seeing how they line up with my batch size