PyTorch基础（八）----- torch.split()方法

一、方法详解

含义：将一个张量分为几个chunks

torch.split(tensor, split_size_or_sections, dim=0)

tensor：要分的张量
split_size_or_sections：
- 如果该项参数的值为一个int类型的value值，那么该方法会将tensor划分为同等数量的张量；如果tensor的size沿着给定的不能整除split_size，那么最后一个chunk相较于其它chunk小；
- 如果是一个list列表，该方法会将tensor划分为len(split_size_or_sections)的张量。
dim：划分张量所依据的维度
return：返回的是一个tuple

这样纯靠文字解释，是很抽象的，我们直接通过以下的案例来对这个方法进行彻底的掌握。

二、案例

案例1

 import torch# 创建一个张量x = torch.arange(10).reshape(5, 2)print(x)

y = torch.split(x, 2)
print(y, type(y))

案例2

x = torch.arange(10).reshape(5, 2)
print(x)
y = torch.split(x, [2, 3])
print(y)

案例3

x = torch.arange(10).reshape(5, 2)
print(x)
y = torch.split(x, [1, 4])
print(y)

相信大家通过以上这3个案例，应该可以掌握Torch.split()这个方法了，最后希望对大家有所帮助！

PyTorch基础（八）----- torch.split()方法相关推荐

3.Pytorch基础模块torch的API之Indexing,Slicing,Joining,Mutating Ops实例详解
文章目录 0. torch 1. Tensors 2. Creation Ops 3. Indexing,Slicing,Joining,Mutating Ops 3.1 torch.cat() 3. ...
pytorch基础-使用 TORCH.AUTOGRAD 进行自动微分（5）
在训练神经网络时,最常用的算法是反向传播.PyTorch的反向传播(即tensor.backward())是通过autograd包来实现的,autograd包会根据tensor进行过的数学运算来自动计 ...
PyTorch基础（15）-- torch.flatten()方法
前言最近在复现论文中一个块的时候需要使用到torch.flatten()这个方法,这个方法其实很简单,但其中有一些细节可能需要注意,且有个关键点很容易忘记,故在此记录以备查阅. 方法解析 flatt ...
PyTorch基础（十）----- torch.max()方法
一.前言这个方法跟上一篇文章的torch.max()方法非常类似,只不过一个是求最大值,一个是求平均值.在某些情况下,甚至可以代替下采样中的最大池化和平均池化,所以说,这两个方法的用处还是蛮大的. ...
PyTorch基础（六）----- torch.eq()方法
一.torch.eq()方法详解对两个张量Tensor进行逐元素的比较,若相同位置的两个元素相同,则返回True:若不同,返回False. torch.eq(input, other, *, out ...
Python基础知识详解从入门到精通（八）魔法方法
目录 Python基础知识详解从入门到精通(八)魔法方法什么是魔法方法基础魔法方法(较为常用) 比较操作符算数运算符反运算(类似于运算方法) 增量赋值运算一元操作符类型转换上下文管理( ...
「Python基础知识」Python的split方法如何使用
Python字符串中的方法有很多,其中split方法是较为常用的方法. 了解一个内建函数/方法比较好的方式,就是使用help()函数查看它的文档字符串,打开Python解释器交互模式,输入: 其中st ...
pytorch基础知识+构建LeNet对Cifar10进行训练+PyTorch-OpCounter统计模型大小和参数量+模型存储与调用
整个环境的配置请参考我另一篇博客.ubuntu安装python3.5+pycharm+anaconda+opencv+docker+nvidia-docker+tensorflow+pytorch+C ...
Pytorch基础学习（第一章-PyTorch基础概念）
课程一览表: 目录一..pytorch简介二.环境配置 1.pycharm 2.annaconda 3.安装pytorch 三.张量简介与创建 1.张量的概念 2.Tensor--直接创建 3. ...

PyTorch基础（八）----- torch.split()方法

一、方法详解

tensor：要分的张量

split_size_or_sections：

dim：划分张量所依据的维度

return：返回的是一个tuple

二、案例

案例1

案例2

案例3

PyTorch基础（八）----- torch.split()方法相关推荐

最新文章

热门文章