Shouldn't really mention neural network here. This is a utility file not necessarily "related" to a neural network imo.

Generalized comment?