Я хочу обрезать все веса в абсолютном диапазоне после каждого обновления весов, что является ключевой операцией в WGAN.
Как мне это сделать в кафе?