diff --git a/mshadow/extension/spatial_upsampling_nearest.h b/mshadow/extension/spatial_upsampling_nearest.h index 3a731b6c899f..534fbdd9ebe0 100644 --- a/mshadow/extension/spatial_upsampling_nearest.h +++ b/mshadow/extension/spatial_upsampling_nearest.h @@ -50,7 +50,7 @@ struct Plan, DType> { : src_(MakePlan(e.src_)), scale_(e.scale_), new_height_(e.shape_[srcdim - 2]), - src_height_(static_cast(e.shape_[srcdim - 2] / 2)) {} + src_height_(static_cast(e.shape_[srcdim - 2] / e.scale_)) {} MSHADOW_XINLINE DType Eval(index_t i, index_t j) const { const index_t x = j; const index_t y = i % new_height_;