예제 #1
0
        /// <summary>次元ごとに最小インデクスを抽出</summary>
        public static VariableNode ArgMin(VariableNode x)
        {
            Function function = new Functions.Indexer.ArgMin();

            VariableNode y = Apply(function, x)[0];

            return(y);
        }
예제 #2
0
        /// <summary>次元ごとに最小インデクスを抽出</summary>
        public static Tensor ArgMin(Tensor x)
        {
            Function function = new Functions.Indexer.ArgMin();

            Tensor y = new Tensor(Shape.Vector(x.Batch));

            function.Execute(new Tensor[] { x }, new Tensor[] { y });

            return(y);
        }