com.komputation.cuda.layers.forward.activation.CudaSigmoidLayer.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.layers.forward.activation

import com.komputation.cuda.kernels.Kernel

class CudaSigmoidLayer internal constructor(
    name : String? = null,
    numberRows : Int,
    numberColumns : Int,
    createForwardKernel: () -> Kernel,
    createBackwardKernel: () -> Kernel,
    warpSize : Int,
    maximumNumberThreadsPerBlock : Int) : BaseCudaEntrywiseActivationLayer(name, createForwardKernel, createBackwardKernel, numberRows, numberColumns, warpSize, maximumNumberThreadsPerBlock)