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)

返回：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]

相关用法

注：本文由纯净天空筛选整理自Jitender_1998大神的英文原创作品 Tensorflow | tf.data.Dataset.from_tensor_slices()。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。