BatchNorm

case class BatchNorm(scope: Scope, input: Variable, weight: Variable, bias: Variable, runningMean: STen, runningVar: STen, training: Boolean, momentum: Double, eps: Double) extends Op
trait Serializable
trait Product
trait Equals
trait Op
class Object
trait Matchable
class Any

Value members

Inherited methods

def productElementNames: Iterator[String]
Inherited from:
Product
def productIterator: Iterator[Any]
Inherited from:
Product

Concrete fields

val expectedShape: List[Long]
val input_flattened: Tensor
val output: Tensor
val output_reshaped: Tensor
override val params: List[(Variable, (STen, STen) => Unit)]
val saveInvstd: Tensor
val saveMean: Tensor
override val value: Variable