コード例 #1
0
        public Tensor Argmax(Tensor result, Tensor src, int dimension)
        {
            Tensor writeTarget = NativeWrapper.CreateResultDimensionwise(result, src, dimension);

            TensorApplyCPU.Argmax(writeTarget, src, dimension);

            return(writeTarget);
        }