Exemplos de TensorShader.Functions.Indexer ArgMin em C# (CSharp)

Exemplo n.º 1

Arquivo: ArgMin.cs Projeto: tk-yoshimura/TensorShaderAVX

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

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

            return(y);
        }

Exemplo n.º 2

Arquivo: ArgMin.cs Projeto: tk-yoshimura/TensorShaderAVX

        /// <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);
        }