public Tensor Argmin(Tensor x, int dim = 0) { dim = CorrDim(x.DimCount, dim); Tensor result = Out(Algorithm.TopK(In(x), 1, 0, 1)); return(result); }
public Tensor Argmin(Tensor x, int dim = 0) { if (dim < 0) { dim = CorrDim(x.DimCount, dim); } if (dim == 1) { if (x.DimCount == 2) { x = x.Transpose(); } else if (x.DimCount == 3) { x = x.Transpose(0, 2, 1); } else if (x.DimCount == 4) { x = x.Transpose(0, 1, 3, 2); } } Tensor result = Out(Algorithm.TopK(In(x), 1, 0, 1)); if (dim == 1) { if (result.DimCount == 2) { result = result.Transpose(); } else if (result.DimCount == 3) { result = result.Transpose(0, 2, 1); } else if (result.DimCount == 4) { result = result.Transpose(0, 1, 3, 2); } } return(result); }