Use numpy.pad to pad input