MCPcopy Create free account
hub / github.com/Raibows/MLinRust / argmax

Function argmax

src/ndarray/utils.rs:119–132  ·  view source on GitHub ↗

non-inplace operation: argmax

(x: &NdArray, dim: i32)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 3

predictMethod · 0.85
test_argmaxFunction · 0.85
mainFunction · 0.85

Calls 1

gather_by_a_specific_dim_and_doFunction · 0.85

Tested by 1

test_argmaxFunction · 0.68