Bug Report for https://neetcode.io/problems/handwritten-digit-classifier The output tensor values don't match for a simple feed forward NN implementation