张量流 | tf.data.Dataset.from_tensor_slices()
借助tf.data.Dataset.from_tensor_slices()
方法,我们可以使用tf.data.Dataset.from_tensor_slices()
方法以对象的形式获取数组的切片。
Syntax : tf.data.Dataset.from_tensor_slices(list)
Return : Return the objects of sliced elements.
示例 #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]