block
block copied to clipboard
Add a `dim`/`axis` arg to support batched tensors