ParametrizedTruncatedNormal

class ParametrizedTruncatedNormal(mean: Float, stdev: Float, p1: Float, p2: Float, seed: Long) : Initializer

Initializer that generates a parametrized truncated normal distribution.

Constructors

ParametrizedTruncatedNormal
Link copied to clipboard
fun ParametrizedTruncatedNormal(mean: Float = 0.0f, stdev: Float = 1.0f, p1: Float = -10.0f, p2: Float = 10.0f, seed: Long)

Creates a ParametrizedTruncatedNormal initializer.

Functions

apply
Link copied to clipboard
fun apply(fanIn: Int, fanOut: Int, tf: Ops, input: Operand<Float>, name: String): Assign<Float>

Adds an Assign Op to the graph to initialize a tensorflow variable as specified by the initializer.

initialize
Link copied to clipboard
open override fun initialize(fanIn: Int, fanOut: Int, tf: Ops, shape: Operand<Int>, name: String): Operand<Float>

Returns a Tensor object initialized as specified by the initializer.

toString
Link copied to clipboard
open override fun toString(): String