跳转至

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】



回到顶部