patrick-kidger/torchtyping · AI Repo StatsPApatrick-kidger/torchtyping
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
Training & fine-tuningPython Apache-2.0 44created 2021-03-28 #named-tensors#python-typing#pytorch#shape#tensors#typing