make sure partial predictions match rensor shape #31

Closed pkopper closed 8 months ago