/// <summary>次元ごとに最大インデクスを抽出</summary> public static VariableNode ArgMax(VariableNode x) { Function function = new Functions.Indexer.ArgMax(); VariableNode y = Apply(function, x)[0]; return(y); }
/// <summary>次元ごとに最大インデクスを抽出</summary> public static Tensor ArgMax(Tensor x) { Function function = new Functions.Indexer.ArgMax(); Tensor y = new Tensor(Shape.Vector(x.Batch)); function.Execute(new Tensor[] { x }, new Tensor[] { y }); return(y); }