码迷,mamicode.com
首页 > 其他好文 > 详细

tf.split函数的用法(tensorflow1.13.0)

时间:2019-08-25 17:56:42      阅读:139      评论:0      收藏:0      [点我收藏+]

标签:inpu   一个   input   函数   split函数   int   切割   dimen   ssi   

tf.split(input, num_split, dimension):

dimension指输入张量的哪一个维度,如果是0就表示对第0维度进行切割;num_split就是切割的数量,如果是2就表示输入张量被切成2份,每一份是一个列表。

 例如:

技术图片
import tensorflow as tf;
import numpy as np;

A = [[1,2,3],[4,5,6]]
x = tf.split(A, 3, 1)

with tf.Session() as sess:
        c = sess.run(x)
        for ele in c:
                print( ele )
技术图片

 输出:

[[1]
 [4]]
[[2]
 [5]]
[[3]
 [6]]

tf.split函数的用法(tensorflow1.13.0)

标签:inpu   一个   input   函数   split函数   int   切割   dimen   ssi   

原文地址:https://www.cnblogs.com/HYWZ36/p/11408451.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!