layers

Type Members

1. abstract class Activation extends Layer[ops.Output, ops.Output]

   

2. case class AddBias(name: String, initializer: Initializer = RandomNormalInitializer()) extends Layer[ops.Output, ops.Output] with Product with Serializable

   

3. case class AudioSummary(name: String, tag: String, samplingRate: tensors.Tensor, maxOutputs: Int = 3, family: String = null, collections: Set[Key[ops.Output]] = Set(Graph.Keys.SUMMARIES)) extends Summary with Product with Serializable

   

4. case class CReLU(name: String) extends Activation with Product with Serializable

   

5. case class Cast(name: String, dataType: types.DataType) extends Layer[ops.Output, ops.Output] with Product with Serializable

   

6. case class Compose[T, R, S](name: String, layer1: Layer[T, R], layer2: Layer[R, S]) extends Layer[T, S] with Product with Serializable

   

7. case class Concatenate[T, R](name: String, layers: Seq[Layer[T, R]]) extends Layer[T, Seq[R]] with Product with Serializable

   

8. case class Conv2D(name: String, filterShape: Shape, stride1: Long, stride2: Long, padding: PaddingMode, dataFormat: CNNDataFormat = CNNDataFormat.default, useCuDNNOnGPU: Boolean = true, weightsInitializer: Initializer = RandomNormalInitializer()) extends Layer[ops.Output, ops.Output] with Product with Serializable

   

9. case class Dropout(name: String, keepProbability: Float, noiseShape: Shape = null, seed: Option[Int] = None) extends Layer[ops.Output, ops.Output] with Product with Serializable

   

10. case class ELU(name: String) extends Activation with Product with Serializable

    

11. case class Embedding(name: String, vocabularySize: Int, embeddingSize: Int, dataType: types.DataType, partitionStrategy: ops.Embedding.PartitionStrategy = ops.Embedding.ModStrategy, transformFn: (ops.Output) ⇒ ops.Output = null, maxNorm: tensors.Tensor = null) extends Layer[ops.Output, ops.Output] with Product with Serializable

    

12. case class Flatten(name: String) extends Layer[ops.Output, ops.Output] with Product with Serializable

    

13. case class HistogramSummary(name: String, tag: String, family: String = null, collections: Set[Key[ops.Output]] = Set(Graph.Keys.SUMMARIES)) extends Summary with Product with Serializable

    

14. case class Identity[T](name: String) extends Layer[T, T] with Product with Serializable

    

15. case class ImageSummary(name: String, tag: String, badColor: tensors.Tensor = Tensor(UINT8, 255, 0, 0, 255), maxOutputs: Int = 3, family: String = null, collections: Set[Key[ops.Output]] = Set(Graph.Keys.SUMMARIES)) extends Summary with Product with Serializable

    

16. case class Input[T, O, DA, D, S](_dataType: DA, shape: S, name: String = "Input")(implicit evDAToD: Aux[DA, D], evDToO: Aux[D, O], evOToT: Aux[O, T], evData: Aux[T, O, D, S]) extends Product with Serializable

    

17. case class L2Loss(name: String) extends Loss[(ops.Output, ops.Output)] with Product with Serializable

    

18. abstract class Layer[T, R] extends AnyRef

    

19. case class Linear(name: String, units: Int, useBias: Boolean = true, weightsInitializer: Initializer = RandomNormalInitializer(), biasInitializer: Initializer = RandomNormalInitializer()) extends Layer[ops.Output, ops.Output] with Product with Serializable

    

20. case class LogPoissonLoss(name: String, computeFullLoss: Boolean = false) extends Loss[(ops.Output, ops.Output)] with Product with Serializable

    

21. case class LogSoftmax(name: String) extends Layer[ops.Output, ops.Output] with Product with Serializable

    

22. abstract class Loss[T] extends Layer[T, ops.Output]

    

23. case class Map[T, R, S, CC[A] <: TraversableLike[A, CC[A]]](name: String, layer: Layer[CC[T], CC[R]], mapLayer: Layer[R, S])(implicit cbfRS: CanBuildFrom[CC[R], S, CC[S]]) extends Layer[CC[T], CC[S]] with Product with Serializable

    

24. case class MaxPool(name: String, windowSize: Seq[Long], stride1: Long, stride2: Long, padding: PaddingMode, dataFormat: CNNDataFormat = CNNDataFormat.default) extends Layer[ops.Output, ops.Output] with Product with Serializable

    

25. case class Mean(name: String) extends Layer[ops.Output, ops.Output] with Product with Serializable

    

26. case class OneHot(name: String, numberOfLabels: Int) extends Layer[ops.Output, ops.Output] with Product with Serializable

    

27. case class ReLU(name: String, alpha: Float = 0.0f) extends Activation with Product with Serializable

    

28. case class ReLU6(name: String) extends Activation with Product with Serializable

    

29. case class SELU(name: String) extends Activation with Product with Serializable

    

30. case class ScalarSummary(name: String, tag: String, family: String = null, collections: Set[Key[ops.Output]] = Set(Graph.Keys.SUMMARIES)) extends Summary with Product with Serializable

    

31. case class SequenceLoss(name: String, averageAcrossTimeSteps: Boolean = true, averageAcrossBatch: Boolean = true, weights: tensors.Tensor = null, lossFn: (ops.Output, ops.Output) ⇒ ops.Output = ...) extends Loss[(ops.Output, ops.Output)] with Product with Serializable

    

32. case class Sigmoid(name: String) extends Activation with Product with Serializable

    

33. case class SigmoidCrossEntropy(name: String) extends Loss[(ops.Output, ops.Output)] with Product with Serializable

    

34. case class Softmax(name: String) extends Layer[ops.Output, ops.Output] with Product with Serializable

    

35. case class SoftmaxCrossEntropy(name: String) extends Loss[(ops.Output, ops.Output)] with Product with Serializable

    

36. case class Softplus(name: String) extends Activation with Product with Serializable

    

37. case class Softsign(name: String) extends Activation with Product with Serializable

    

38. case class SparseSoftmaxCrossEntropy(name: String) extends Loss[(ops.Output, ops.Output)] with Product with Serializable

    

39. case class Squeeze(name: String, axes: Seq[Int] = null) extends Layer[ops.Output, ops.Output] with Product with Serializable

    

40. case class Sum(name: String) extends Layer[ops.Output, ops.Output] with Product with Serializable

    

41. abstract class Summary extends Layer[ops.Output, ops.Output]

    

42. case class Transpose(name: String, permutation: Seq[Int]) extends Layer[ops.Output, ops.Output] with Product with Serializable