torch.aminmax
function aminmax<D extends DType = DType, Dev extends DeviceType = DeviceType>(input: Tensor<Shape, D, Dev>, options?: AminmaxOptions): { min: Tensor<Shape, D, Dev>; max: Tensor<Shape, D, Dev> }function aminmax<D extends DType = DType, Dev extends DeviceType = DeviceType>(input: Tensor<Shape, D, Dev>, dim: number, keepdim: boolean): { min: Tensor<Shape, D, Dev>; max: Tensor<Shape, D, Dev> }Returns both the minimum and maximum values of the input tensor.
Parameters
optionsAminmaxOptionsoptional
Returns
Examples
const x = torch.tensor([[1, 5], [3, 2]]);
const { min, max } = torch.aminmax(x, 1);
// min: [1, 2], max: [5, 3]