I trained a CNN on an MNIST dataset, and planned to use it for inference with tf.js.
However, when checking the model\'s gradients for one of the test images, I notic