skainet-compile-core/sk.ainet.tape/GradientTape/computeGradients

computeGradients

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

Compute gradients for the recorded operations

abstract val computeGradients: Boolean(source)

Whether to compute gradients for recorded operations