Python Count Unique Values In Numpy Array To Tensor