Closed wiseodd closed 2 months ago
Fixes #235
Rationale on why we don't want to throw an error:
model(x).squeeze()
y.shape == model(x).shape[0:-1]
model(x)
y.shape == model(x).shape
y.shape
Fixes #235
Rationale on why we don't want to throw an error:
model(x).squeeze()
withy.shape == model(x).shape[0:-1]
model(x)
withy.shape == model(x).shape
y.shape
's. It's too much to ask to requirey.shape == model(x).shape
since PyTorch doesn't.