make sure partial predictions match rensor shape #30

Closed pkopper closed 8 months ago