pytc.kmat._pad_axis

pytc.kmat._pad_axis(arr, axis, pad)[source]

Pad arr with zeros by pad along axis. Works for np or jnp.