Open source
Specifying tensor shapes using Python type annotations

TensorAnnotations is a library enabling annotation of tensor shapes using Python type annotations. This enables shape annotations to be checked statically by a standard Python type-checker - catching a range of bugs caused by e.g. wrong selection or reduction of axes before you run your code (even when the errors would not necessarily throw a runtime exception!)

No items found.