Solved by jmandivarapu1 in post #2 
              
              
                Usually, predicted returns a tensor array. You can iterate through that array and check for each item. I assume you can convert it into .numpy() and use the parallel method to do it more quickly