Self-Attention Matrix Access for ViTAutoEnc and UneTr

[a-parida12]

The self-attention matrix of the ViT has been made accessible while resolving issue #6032. Changes in the downstream classes are required to be updated to be able to access the feature.

#6032 (comment)