small change to connection prediction output
created tensor utils module