GradientTape

Gradient tape for automatic differentiation support

Properties

Whether to compute gradients for recorded operations

abstract fun <T : DType, V> computeGradients(targets: List<Tensor<T, V>>, sources: List<Tensor<T, V>>): Map<Tensor<T, V>, Tensor<T, V>>

Compute gradients for the recorded operations

abstract fun <T : DType, V> stopWatching(tensors: List<Tensor<T, V>>)

Stop watching tensors

abstract fun <T : DType, V> watch(tensors: List<Tensor<T, V>>)

Watch tensors for gradient computation