I guess having the user transform them into numpy before comparison is OK for now, and to be consistent I’ll make both data and labels tvm.runtime.ndarrays
. I can put a note in the documentation that they need to convert them to numpy arrays before comparing them.
It would be nice if there was a way to directly compare the values of tvm.runtime.ndarrays
, though.