Merge pull request #1160 from streamjoin/update-type-tensor

Update the data types and tensor operations for training