• Home
• com.thoughtworks.deeplearning
• double_2.10
• 1.0.0-M3
• source code
• LessThan.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.LessThan.scala Maven / Gradle / Ivy

package com.thoughtworks.deeplearning
package double.layers

import cats._
import cats.implicits._
import com.thoughtworks.deeplearning.BufferedLayer
import com.thoughtworks.deeplearning.boolean.utilities.BooleanMonoidBatch

/**
  * @author 杨博 (Yang Bo) <[email protected]>
  */
final case class LessThan[Input0 <: Batch](
    operand1: Layer.Aux[Input0, Batch.Aux[Eval[scala.Double], Eval[scala.Double]]],
    operand2: Layer.Aux[Input0, Batch.Aux[Eval[scala.Double], Eval[scala.Double]]]
) extends BufferedLayer.Binary {

  type BufferedBatch = BooleanMonoidBatch with MonoidBatch with BinaryBatch

  type Input = Input0

  override protected def rawForward(input0: Input): BufferedBatch = {
    new {
      override final val input = input0
    } with BooleanMonoidBatch with MonoidBatch with BinaryBatch {
      override val value = upstream1.value.map2(upstream2.value)(_ < _).memoize

      override protected def rawBackward(delta: Eval[scala.Boolean]): Unit = {
        upstream1.backward(Eval.now(0.0))
        upstream2.backward(Eval.now(0.0))
      }
    }
  }
}