Documentation

Creates a new, uninitialized variable.

Gets the value stored in a variable.

Note that this op is stateful since it depends on the value of the variable; however, it may be CSE'd with other reads in the same context. The context can be fixed by using render along with (for example) withControlDependencies. For example:

  runSession $ do
    v <- variable []
    a <- assign v 24
    r <- withControlDependencies a $ render $ readValue v + 18
    result <- run r
    liftIO $ (42 :: Float) @=? unScalar result

The initial value of a Variable created with initializedVariable.

Creates a variable initialized to the given value. Initialization happens next time session runs.

Creates a zero-initialized variable with the given shape.

Sets the value of a variable.

Increments the value of a variable.

Update '*var' according to the Adam algorithm.

lr_t <- learning_rate * sqrt(1 - beta2^t) / (1 - beta1^t) m_t <- beta1 * m_{t-1} + (1 - beta1) * g_t v_t <- beta2 * v_{t-1} + (1 - beta2) * g_t * g_t variable <- variable - lr_t * m_t / (sqrt(v_t) + epsilon)