📌  相关文章
📜  张量流 | tf.data.Dataset.from_tensor_slices()

📅  最后修改于: 2022-05-13 01:54:44.430000             🧑  作者: Mango

张量流 | tf.data.Dataset.from_tensor_slices()

借助tf.data.Dataset.from_tensor_slices()方法,我们可以使用tf.data.Dataset.from_tensor_slices()方法以对象的形式获取数组的切片。

示例 #1:
在这个例子中我们可以看到,通过使用tf.data.Dataset.from_tensor_slices()方法,我们可以得到列表或数组的切片。

# import tensorflow
import tensorflow as tf
  
# using tf.data.Dataset.from_tensor_slices() method
gfg = tf.data.Dataset.from_tensor_slices([1, 2, 3, 4, 5])
  
for ele in gfg:
    print(ele.numpy())

输出 :

示例 #2:

# import tensorflow
import tensorflow as tf
  
# using tf.data.Dataset.from_tensor_slices() method
gfg = tf.data.Dataset.from_tensor_slices([[5, 10], [3, 6]])
  
for ele in gfg:
    print(ele.numpy())

输出 :