• Home
• com.thoughtworks.deeplearning
• double_2.10
• 1.0.0-M3
• source code
• Weight.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.thoughtworks.deeplearning.double.layers.Weight.scala Maven / Gradle / Ivy

package com.thoughtworks.deeplearning.double.layers

import cats._
import cats.implicits._
import com.thoughtworks.deeplearning.double.optimizers.Optimizer
import com.thoughtworks.deeplearning.{Batch, Layer}
import com.thoughtworks.deeplearning.double.utilities.DoubleMonoidBatch

/**
  * @author 杨博 (Yang Bo) <[email protected]>
  */
final case class Weight(var rawValue: scala.Double)(implicit optimizer: Optimizer)
    extends Layer
    with DoubleMonoidBatch {
  override type Input = Batch
  override type Output = Batch.Aux[Data, Delta]

  override def addReference() = this

  override def forward(any: Input) = this

  override def backward(delta: Delta): Unit = {
    synchronized {
      rawValue = optimizer.updateDouble(rawValue, delta.value)
    }
  }

  override def value = Eval.now(rawValue)

  override def close(): Unit = {}

}