improve lax.pad shape rule (#4234)

Commit

5 years ago

improve lax.pad shape rule (#4234) It's now: * better tested * better at catching errors * faster * easier to read