I am also interested in a function like this.  Functions like mean(), median(), and mode() can take in a list of dimensions to reduce; is functionality for other functions like min() and max() possible?
My use case is that I want to quantize a weight tensor; if I am interested in quantizing using constants from the whole tensor, a step along the way is to calculate the min and max of the whole tensor, which min() and max() do accomplish.  However, I want to compare this against channel-wise quantization (with different constants for zero point, scale, etc), and it would be great if min() and max() produced the right outputs given a list of dimensions to reduce.