jatentaki/torch-dimcheck

Weird breakage

Opened this issue · 0 comments

jatentaki commented 
from torch_dimcheck import dimchecked

@dimchecked
def f(t: ['B', 'N', 3]) -> ['B', 'N', 3]:
    return t

f(torch.randn(2, 3))