Wrapping your code with tf.device('/cpu:0'): I get 9.63 seconds (M1) and with GPU 60s.
Topic:
Machine Learning & AI
SubTopic:
General
Tags: