Commit d05d894
committed
Reject sub-2D channel-first tensors in _normalize_spatial_ndim
Raise a clear ValueError when a channel-first tensor has fewer than 2
dims instead of silently clamping spatial_ndim to 1, per review feedback.
Signed-off-by: Soumya Snigdha Kundu <soumya_snigdha.kundu@kcl.ac.uk>1 parent 9f8d9f6 commit d05d894
1 file changed
Lines changed: 4 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
37 | 41 | | |
38 | 42 | | |
39 | 43 | | |
| |||
0 commit comments