Можем ли мы изменить данные тензора с помощью потока управления вместо операций тензора при сохранении графика?

#pytorch

Вопрос:

В моей функции потерь с вводом x (2-d массив с плавающей точкой) я пытаюсь использовать пользовательскую логику, например, замену статических индексов на логику потока управления для отдельных значений с плавающей точкой, чтобы изменить данные тензора в x перед передачей их через двоичную потерю перекрестной энтропии.

Однако для этого я попытался создать новый тензор с require_grad и измененными данными, однако тензоры вывода потерь, вероятно, больше не отображаются на графике