so

def get_data(): test_data_shape = [3,2,2] out = np.random.randint(100, 255, size=test_data_shape) print(out) print("***----***") return outdef test_cpu(): #prefetch_queue_depth 表示缓存多少个batch_size的数据在内存,默认是2 .