wenkao123456的博客

这个函数就是将输入的张量，按第一维度切分成多个小的张量。代码如下：import tensorflow as tfb = tf.random.normal([3, 1, 2, 1])bb = tf.data.Dataset.from_tensor_slices(b)print(bb)print('-----------------------')for i in bb: ...