Inconsistent data types (numpy / torch)

[mhubii]

A lot of the legacy code here uses numpy, an effort should be made to exclusively use torch containers.