It might be useful to extend the current clamp() function, where the min and max can be tensor.

e.g. To clip the value for a [5, 3]-tensor, we could define min and max for each element, or each row/column.

Finally, it might be good to rename it to clip to be consistent with numpy.clip

cc @ezyang @gchanan @zou3519 @bdhirsh @jbschlosser @mruberry @rgommers @heitorschueroff