com.komputation.cuda.loss.CudaLossFunction.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.cuda.loss

import jcuda.Pointer
import com.komputation.layers.Resourceful

interface CudaLossFunction : Resourceful {

    fun accumulate(pointerToPredictions: Pointer, pointerToTargets : Pointer, batchSize: Int)

    fun accessAccumulation() : Float

    fun backward(pointerToPredictions: Pointer, pointerToTargets : Pointer, batchSize : Int): Pointer

}