tf.split

tf.split(
    value,
    num_or_size_splits,
    axis=0,
    num=None,
    name='split'
)

把张量分解成子张量。如果num_or_size_split是一个整数,那么值将沿着维度轴拆分为num_split较小的张量。这要求num_split均匀地分割value.shape[axis]。如果num_or_size_split是一个一维张量(或列表),称之为size_split,值被分割成len(size_split)元素。第i个元素的形状与值的大小相同,除了尺寸为size_split [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: 要分割的张量。
  • num_or_size_split: 要么是一个整数,表示沿split_dim进行的分割的数量,要么是一个一维整数张量或Python列表,其中包含沿split_dim进行的每个输出张量的大小。如果一个标量,那么它必须均匀地除value.shape[axis];否则,拆分维度上的大小总和必须与值的大小相匹配。
  • axis: 整数或标量int32张量。要分割的维度。必须在[-rank(value), rank(value)]范围内。默认值为0。
  • num: 可选的,用于指定不能从size_split的形状推断输出的数量。
  • name:操作的名称(可选)。

返回值:

  • 如果num_or_size_split是一个标量,则返回num_or_size_split张量对象;如果num_or_size_split是一个一维张量,则返回num_or_size_split。由值分裂得到的张量对象。

异常:

  • ValueError: If num is unspecified and cannot be inferred.

原链接: https://tensorflow.google.cn/versions/r1.14/api_docs/python/tf/split

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 虚拟机安装ubuntu

    公司的电脑是windows的。因为平常要用到linux的独有特性。比如swoole等等。所以,我在虚拟机中安装ubuntu。

    魔王卷子
  • Ubuntu 16.04 Cuda8.0 tensorflow-gpu

    ubuntu 16.04 python 2.7 cuda7.5/Cuda8.0 tensorflow-gpu

    ke1th
  • Docker常用命令

    首先对容器做变更:举个例子 创建容器,启动ubuntu 开启bash 创建一个文件

    lilugirl
  • 使用Typescript开发node.js项目

    typescript 通过构造函数的参数直接定义属性,为了证明,我们创建一个info.ts文件 内容如下

    lilugirl
  • Ubuntu16.0.4+1080ti+ cuda9.1+cudnn7.0.5 安装 及 caffe 测试

    版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zhangjunhit/article/de...

    用户1148525
  • linux 系统网络常见网卡配置

    /etc/sysconfig/network-scripts/ifcfg-eth0

    晴天Online
  • DockerSwarm 微服务部署

    之前《服务Docker化》中,使用 docker-compose.yml 来一次配置启动多个容器,在 Swarm 集群中也可以使用 compose 文件 (do...

    程序员果果
  • 08-可视化操作-探索泰坦尼克灾难数据

    import pandas as pd import numpy as np import matplotlib.pyplot as plt import se...

    光点神奇
  • Hexo让博客梦变的简单

    这一步我不想展开太多,因为真的很简单,关于GIT的配置我在之前的文章中已经提到过了,看官可以查看历史文章

    迹_Jason
  • Shell 之 read脚本输入

    read命令接收标准输入(键盘)的输入,或其他文件描述符的输入(后面在说)。得到输入后,read命令将数据放入一个标准变量中

    晴天Online

扫码关注云+社区

领取腾讯云代金券