diffusion_rs_common::core

Module backprop

source

Structsยง

  • GradStore
    A store for gradients, associating a tensor id to the corresponding gradient tensor, used for back propagation.