com.komputation.cpu.functions.MaximumIndices.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of komputation Show documentation

Komputation is a neural network framework for the JVM written in the Kotlin programming language.

There is a newer version: 0.12.5

package com.komputation.cpu.functions

fun findMaxIndicesInRows(input: FloatArray, numberRows : Int, numberColumns : Int, result : IntArray) {

    for (indexRow in 0 until numberRows) {

        var maxValue = input[indexRow]
        var maxIndex = indexRow

        // Go through the columns in the given row
        for (indexColumn in 1 until numberColumns) {

            val index = indexRow + indexColumn * numberRows

            val entry = input[index]

            if (entry > maxValue) {

                maxValue = entry
                maxIndex = index

            }

        }

        result[indexRow] = maxIndex

    }

}

fun findMaxIndex(input: FloatArray, offset : Int, limit : Int): Int {

    var maxIndex = -1
    var maxValue = Float.NEGATIVE_INFINITY

    for(index in offset until offset + limit) {

        val value = input[index]

        if (value > maxValue) {

            maxValue = value
            maxIndex = index

        }

    }

    return maxIndex - offset

}

fun findMaxIndices(input: FloatArray, numberIndices: Int, length: Int, result: IntArray) {

    for (step in 0 until length) {

        val offset = step * numberIndices

        result[step] = findMaxIndex(input, offset, numberIndices)

    }


}