tf.split函数的用法

UESTC_C2_403

于 2017-06-16 19:01:59 发布

阅读量4.3w

点赞数 4

CC 4.0 BY-SA版权

分类专栏： tensorflow用法

本文链接：https://blog.youkuaiyun.com/UESTC_C2_403/article/details/73350457

tensorflow用法专栏收录该内容

64 篇文章

订阅专栏

本文介绍如何使用TensorFlow的tf.split函数对张量进行分割。通过实例演示了指定维度和分割数量的具体操作方法，并提醒读者注意不同版本的TensorFlow中此函数可能存在的差异。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tf.split(dimension, num_split, input)：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(1, 3, A)

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

输出：

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

注意：这个程序安装的tf版本是0.12.0，版本不同会有改动的，也就是函数用法会不同，注意一下子。