I was having trouble seems it seems you can\'t edit a tensor directly or simply convert it to numpy and edit it in that form during training either. To some extent what I\'m