tf.split
2024-08-28 00:01:17
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]]
TensorFlow函数:tf.split
由 Carrie 创建, 最后一次修改 2018-03-15
tf.split函数
split(
value,
num_or_size_splits,
axis=0,
num=None,
name='split'
)
定义在:tensorflow/python/ops/array_ops.py。
参见指南:张量变换>切割和连接
将张量分割成子张量。
如果 num_or_size_splits 是整数类型,num_split,则 value 沿维度 axis 分割成为 num_split 更小的张量。要求 num_split 均匀分配 value.shape[axis]。
如果 num_or_size_splits 不是整数类型,则它被认为是一个张量 size_splits,然后将 value 分割成 len(size_splits) 块。第 i 部分的形状与 value 的大小相同,除了沿维度 axis 之外的大小 size_splits[i]。
例如:
# 'value' is a tensor with shape [5, 30]
# Split 'value' into 3 tensors with sizes [4, 15, 11] along dimension 1
split0, split1, split2 = tf.split(value, [4, 15, 11], 1)
tf.shape(split0) # [5, 4]
tf.shape(split1) # [5, 15]
tf.shape(split2) # [5, 11]
# Split 'value' into 3 tensors along dimension 1
split0, split1, split2 = tf.split(value, num_or_size_splits=3, axis=1)
tf.shape(split0) # [5, 10]
函数参数:
- value:要分割的 Tensor。
- num_or_size_splits:指示沿 split_dim 分割数量的 0-D 整数 Tensor 或包含沿 split_dim 每个输出张量大小的 1-D 整数 Tensor ;如果为一个标量,那么它必须均匀分割 value.shape[axis];否则沿分割维度的大小总和必须与该 value 相匹配。
- axis:A 0-D int32 Tensor;表示分割的尺寸;必须在[-rank(value), rank(value))范围内;默认为0。
- num:可选的,用于指定无法从 size_splits 的形状推断出的输出数。
- name:操作的名称(可选)。
函数返回值:
如果 num_or_size_splits 是标量,返回 num_or_size_splits Tensor对象;如果 num_or_size_splits 是一维张量,则返回由 value 分割产生的 num_or_size_splits.get_shape[0] Tensor对象。
函数可能引发的异常:
- ValueError:如果 num 没有指定并且无法推断。
最新文章
- 混合使用UITabBarController和UINavigationController
- C#多任务并行阶段控制—— Threading.Barrier
- StrongSwan 5.1.1 发布,Linux 的 IPsec 项目
- HTML5第二节
- 读javascript高级程序设计06-面向对象之继承
- 一张图看Goodle Clean设计架构
- 通过Roslyn构建自己的C#脚本
- SGU185 - Two Shortest
- 如何创建WIN服务
- Delphi5 update1的序列号
- MC34063中文资料及应用实例(转)
- 不知道Linux的mysql的root密码
- NHibernate3剖析:Configuration篇之SessionFactory lambda配置
- mock测试之powermock
- [物理学与PDEs]第3章习题1 只有一个非零分量的磁场
- python3-可变长度参数函数(*args 和 **kwargs)
- poj2083 Fractal
- Java学习过程中要记录的地方--汇总
- 前端JavaScript获取时间戳
- golang 的 buffered channel 及 unbuffered channel