tensorflow:tf.stack的用法

来源:互联网 发布:caffe 图像预处理 编辑:程序博客网 时间:2024/05/16 17:48

堆叠list类型的数据:

import tensorflow as tflist_ = [[1, 2, 3], [4, 5, 6]]print(type(list_))a = tf.stack(list_, 0)print(a.get_shape().as_list())
输出结果:

<type 'list'>[2, 3]

原创粉丝点击