It seems as if pytorch tensor operations can preform some tasks faster than numpy array operations. In order to check this we will be switching the numpy operations into equivalent pytorch operations.