tf.split函数的用法

本文介绍如何使用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,版本不同会有改动的,也就是函数用法会不同,注意一下子。

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值