Here is the result below, but one thing I notice is if I run this is gpu it works, and if I run this in cpu, it doesn’t. Do you have any suggestions?

Screen Shot 2021-12-07 at 9.36.38 AM1384×912 106 KB