对两个flatten函数的探究

简介: 对两个flatten函数的探究

1 问题

对torch.nn.flatten()和torch.flatten()两个函数的理解。


2 方法

对于torch.nn.Flatten():
其默认参数为start_dim = 1 , end_dim = -1,即从第1维(第0维不变)开始到最后一维结束将每个batch拉伸成一维:


当仅设置一个参数时,该参数表示 start_dim 的值,即从该维度开始到最后一个维度结束,将每个batch拉伸成一维,其余维度不变:


当设置两个参数时,两个参数分别表示开始维度和结束维度:


Torch.nn.flatten()函数官方文档:

对于torch.flatten():
torch.flatten()函数默认start_dim = 0 , 其余与torch.nn.flatten()相同。


torch.flatten()函数官方文档:

3 结语

通过对照实验,对两个函数的参数进行比照分析,得出结论。

目录
相关文章
|
5月前
|
机器学习/深度学习 PyTorch 算法框架/工具
PyTorch基础之激活函数模块中Sigmoid、Tanh、ReLU、LeakyReLU函数讲解(附源码)
PyTorch基础之激活函数模块中Sigmoid、Tanh、ReLU、LeakyReLU函数讲解(附源码)
66 0
|
5月前
|
机器学习/深度学习 PyTorch 算法框架/工具
PyTorch深度学习基础之Tensor对象及其应用的讲解及实战(附源码 简单易懂 包括分段 映射 矩阵乘法 随机数等等)
PyTorch深度学习基础之Tensor对象及其应用的讲解及实战(附源码 简单易懂 包括分段 映射 矩阵乘法 随机数等等)
36 1
|
12月前
|
PyTorch 算法框架/工具
Pytorch疑难小实验:Torch.max() Torch.min()在不同维度上的解释
Pytorch疑难小实验:Torch.max() Torch.min()在不同维度上的解释
122 0
|
机器学习/深度学习 PyTorch 算法框架/工具
Pytorch中使用torch.nn模块进行神经网络模型初步构造
Pytorch中使用torch.nn模块进行神经网络模型初步构造
81 0
Pytorch中使用torch.nn模块进行神经网络模型初步构造
|
机器学习/深度学习 算法框架/工具
5分钟入门GANS:原理解释和keras代码实现
5分钟入门GANS:原理解释和keras代码实现
171 0
5分钟入门GANS:原理解释和keras代码实现
|
机器学习/深度学习 存储 并行计算
一个快速构造GAN的教程:如何用pytorch构造DCGAN(下)
一个快速构造GAN的教程:如何用pytorch构造DCGAN
116 0
一个快速构造GAN的教程:如何用pytorch构造DCGAN(下)
|
机器学习/深度学习 存储 PyTorch
一个快速构造GAN的教程:如何用pytorch构造DCGAN(上)
一个快速构造GAN的教程:如何用pytorch构造DCGAN
127 0
一个快速构造GAN的教程:如何用pytorch构造DCGAN(上)
|
存储 机器学习/深度学习 算法
(读Yolo3源码发现的不会的函数)Pytorch常用函数记录-pretrained-torch.nn.Upsample()函数-torch.cat-a.permute-a.view()等
(读Yolo3源码发现的不会的函数)Pytorch常用函数记录-pretrained-torch.nn.Upsample()函数-torch.cat-a.permute-a.view()等
|
人工智能 自然语言处理 Python
TF-IDF:概念与python实现
TF-IDF:概念与python实现
TF-IDF:概念与python实现
|
机器学习/深度学习 缓存 自然语言处理
【40】torch.nn汇总(各类与函数接口的解释说明)
【40】torch.nn汇总(各类与函数接口的解释说明)
333 0
【40】torch.nn汇总(各类与函数接口的解释说明)