• Home
• com.thoughtworks.deeplearning
• plugins-floatliterals_2.12
• 2.2.0-M1
• source code
• FloatLiterals.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.plugins.FloatLiterals.scala Maven / Gradle / Ivy

package com.thoughtworks.deeplearning
package plugins

import com.thoughtworks.deeplearning.DeepLearning.Tape
import com.thoughtworks.raii.asynchronous.Do

import com.thoughtworks.continuation.Continuation

/** A plugin that enables [[scala.Float]] in neural networks. */
trait FloatLiterals {

  trait ImplicitsApi {
    implicit def floatLiteralDeepLearning: DeepLearning.Aux[Float, Float, Float] = new DeepLearning[Float] {
      override type Data = Float
      override type Delta = Float

      override def forward(literal: Float): Do[Tape[Data, Delta]] = {
        Do.now(Tape(literal, Function.const(Continuation.now(()))))
      }
    }
  }

  type Implicits <: ImplicitsApi
}