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
相關文章
相關標籤/搜索