Skip to main content

tf.split用法

按照行或者列分割一个矩阵
第一个参数是待切分的张量，第二个是切分的大小，共切分为三个，每一个分别为4列，15列，11列。第三个参数为1表示按照列切分，为0表示按照行切分。
切分的子块的大小加起来应该等于母块的大小。比如这个例子，4+15+11=30列

看下面的代码例子


Comments

Post a Comment

Popular posts from this blog

Session Run的用法

feed_dict参数的作用是替换图中的某个tensor的值。例如： a = tf.add(2, 5)                        #a=7 b = tf.multiply(a, 3)                 #b=3*7=21 with tf.Session() as sess:     print(sess.run(b))     replace_dict = {a:15}           #用15代替b算式中的a     print(sess.run(b, feed_dict = replace_dict)) --------------------- 输出如下： 21 45
Post a Comment
Read more