Я хочу обучить net с весами, которые суммируются с константой для указанной оси c. Есть ли способ навязать такое условие при тренировках с использованием Keras или Pytorch? Любой указатель приветствуется.