ShapeGuard allows you to very succinctly assert the expected shapes of tensors in a dynamic, einsum inspired way
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool