Tensorflow | TF.data.dataset.from_tensor_slices()
原文:https://www.geesforgeks.org/tensorflow-TF-data-dataset-from_tensor_slices/
借助**tf.data.Dataset.from_tensor_slices()**
方法,我们可以用tf.data.Dataset.from_tensor_slices()
方法得到物体形式的数组切片。
语法:
tf.data.Dataset.from_tensor_slices(list)
返回:返回切片元素的对象。
示例#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())
输出:
1 2 3 4 5
例 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())
输出:
[5,10] 【3,6】