Identity

class Identity(gain: Float) : Initializer

Initializer that generates the identity matrix. This initializer is only usable for generating 2D matrices. Although identity matrices are undefined for non-square matrices an "identity" matrix is generated. E.g. 2 x 3 "identity" matrix ==> [ 1., 0., 0., 0., 1., 0.]

Throws

org.jetbrains.kotlinx.dl.api.core.exception.IdentityDimensionalityException

if is not 2D matrix created.

Constructors

Identity

fun Identity(gain: Float = 1.0f)

Creates a Identity initializer.

Functions

apply

fun apply(fanIn: Int, fanOut: Int, tf: Ops, input: Operand<Float>, name: String): InitializerOperation

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

initialize

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

open override fun toString(): String

Properties

gain

val gain: Float = 1.0f

Identity matrix is multiply by this factor