How do I get the MatMul dimensions for GEMMs used by TensorFlow ( and PyTorch)

AI & Data Science Deep Learning (Training & Inference) Frameworks
1

Hi, TensorBoard shows MatMuls/GEMM ops, but does not show what the tensor dimensions are. Is there way to get those dimensions ?