fixed bug: make prediction with correct dims

Merge request reports

Loading