I’m working with a large PyTorch codebase that has frequent asserts to check that tensors have the expected dtype and shape (all our shapes are static. We don’t use dynamic shapes for anything):