Pytorch chunk

Pytorch chunk的方法做的是对张量进行分块,返回一个张量列表。但如果指定轴的元素个数被chunks除不尽,最后一块的元素个数会少。

torch.chunk(tensor, chunks, dim=0) ->得到一个list的tensors

这个函数的作用是把一个tensor划分到特定数目的块。

chunks的取值 结果
chunks=n 代表切分成几个小块
dim= n 代表要在哪个维度进行操作
数据横向切割图.png

如果竖着切可以看到如下情况

数据竖着切割图.png
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • pytorch torch包 torch 包含了多维张量的数据结构以及基于其上的多种数学操作。 函数 torch....
    NJUST江波阅读 1,067评论 0 0
  • 什么是PyTorch?   PyTorch是Facebook人工智能团队开发的一个机器学习和深度学习工具,用于处理...
    山阴少年阅读 2,152评论 0 13
  • 犹太人简史 犹太人古称希伯来人,其祖先是一个游牧部族,约于公元前2000年从幼发拉底河流域(此为传统说法)来到迦南...
    90a2537c2cd5阅读 436评论 0 1
  • The world as seen in each person's eyes is the manifestat...
    石竹阅读 144评论 0 0
  • 我的一个朋友 她在思念海子 或许 海子心中默写诗句的时候 他的内心正满是悲痛 那颗流血的心 被西伯利亚的冷风裹挟 ...
    是月阅读 213评论 2 1