one_hot_encode should only use torch functions. This would allow the package to be independent of numpy.
one_hot_encode should only use torch functions. This would allow the package to be independent of numpy.