tf.data迭代器问题

import tensorflow as tf import numpy as np def __a(a): b=a+1 b=np.squeeze(b) return b a=np.array(range(5)) b=tf.constant(a) dataset = tf.data.Dataset.from_tensor_slices(a) dataset = d

>>阅读原文<<