use ds.sizes insted of ds.dims

use ds.sizes insted of ds.dims #1454