Imports github.com/Kautenja/gotorch github.com/Kautenja/gotorch/nn/functional Standard library imports fmt image image/draw math unsafe