tf.split()函数

tf.split(
    value,
    num_or_size_splits,
    axis=0,
    num=None,
    name='split'
)
  • value:准备切分的张量
  • num_or_size_splits:准备切成几份
  • axis : 准备在第几个维度上进行切割

其中分割方式分为两种

  1. 如果num_or_size_splits 传入的 是一个整数,那直接在axis=D这个维度上把张量平均切分成几个小张量
  2. 如果num_or_size_splits 传入的是一个向量(这里向量各个元素的和要跟原本这个维度的数值相等)就根据这个向量有几个元素分为几项)

举个例子

# 张量为(5, 30)
# 这个时候5是axis=0, 30是axis=1,如果要在axis=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]
# 传入整数时
split0, split1, split2 = tf.split(value, num_or_size_splits=3, axis=1)
tf.shape(split0)  # [5, 10]

参考:https://blog.csdn.net/mls0311/article/details/82052472 https://www.w3cschool.cn/tensorflow_python/tensorflow_python-x1yu2mcy.html

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

我来说两句

0 条评论
登录 后参与评论

相关文章

  • tf.name_scope()和tf.variable_scope()

    tf.variable_scope可以让变量有相同的命名,包括tf.get_variable得到的变量,还有tf.Variable的变量

    周小董
  • 利用 Python + Selenium 实现对页面的指定元素截图(可截长图元素)

    WebDriver.Chrome自带的方法只能对当前窗口截屏,且不能指定特定元素。若是需要截取特定元素或是窗口超过了一屏,就只能另辟蹊径了。

    周小董
  • tf.expand_dims()

    TensorFlow中,想要维度增加一维,可以使用tf.expand_dims(input, dim, name=None)函数。当然,我们常用tf.resha...

    周小董
  • OpenCV-Mat与Base64之间的相互转换

    PS:不得不吐槽一句这年头WebSocket转MAT这么复杂,C++后端都隐藏实力去了吗?

    Pulsar-V
  • 学界 | 从感知机到GAN,机器学习简史梳理

    选自chatbotnewsdaily 机器之心编译 参与:蒋思源、李亚洲 机器学习是人工智能的一个重要分支,也是如今学界、产业界的热门研究。公司、高校倾倒了许多...

    机器之心
  • 教你用 Python 和 Keras 建立自己的 AlphaZero

    作者David Foster。 翻译 | 黄伟聪 董星 校对 | 凡江 ? 在这篇文章,我会试着阐明三件事: AlphaZero之所以是人工智能前...

    AI研习社
  • 【趣闻】18个世界知名企业的经典段子,你看懂了几个

    这18个世界知名企业的经典段子,虽已流传江湖已久,但放在今日,还能让你在大笑的同时思考职场。

    华章科技
  • 往"某度文库"上传资源之前,请先做好这些...想到一句好玩的话:知道是你干的,只是懒得抓你!如果你喜欢python,喜欢故事,请点赞或关注我!您的支持是对作者最大的鼓励!

    先讲个相关的故事:匿名黑客的"复仇行动" 2010年12月10日,黑客组织匿名者发布了一条消息,解释了他们发起最近一次代号为”复仇行动”的攻击的大致动机(Pr...

    zhaoolee
  • 如何在CDSW中使用GPU运行深度学习

    Fayson
  • Google首席科学家谈Google是怎么做深度学习的

    2016年3月7日,谷歌首席科学家,MapReduce、BigTable等系统的创造者,Jeff Dean受邀韩国大学,演讲主题《大规模深度学习》,这里部分来自...

    代码医生工作室

扫码关注云+社区

领取腾讯云代金券