• Home
• com.thoughtworks.deeplearning
• boolean_2.11
• 1.0.0-M3
• source code
• If.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.boolean.layers.If.scala Maven / Gradle / Ivy

package com.thoughtworks.deeplearning.boolean.layers

import com.thoughtworks.deeplearning.{Batch, Layer}
import com.thoughtworks.deeplearning.Layer._
import com.thoughtworks.deeplearning.boolean.utilities._

/**
  * @author 杨博 (Yang Bo) <[email protected]>
  */
final case class If[Input0 <: Batch, OutputData0, OutputDelta0](
    condition: Layer.Aux[Input0, Boolean#Batch],
    `then`: Layer.Aux[Input0, Batch.Aux[OutputData0, OutputDelta0]],
    `else`: Layer.Aux[Input0, Batch.Aux[OutputData0, OutputDelta0]])
    extends Layer {
  override type Input = Input0
  override type Output = Batch.Aux[OutputData0, OutputDelta0]

  override def forward(input: Input0) = {
    resource.managed(condition.forward(input)).acquireAndGet { conditionBatch =>
      (if (conditionBatch.value.value) `then` else `else`).forward(input)
    }
  }
}