pytorch Flatten展平

经过测试，cpu模式中，用Module的方式比直接在推理中展平平均快1~5ms

from torch import nn
import torch
import mathclass Flatten(Module):def forward(self, input):return input.view(input.size(0), -1)

直接展平：

  out = self.avgpool(out)# out = self.conv_6_flatten(out)out = out.view(out.size(0), -1)

展平为向量

label = label.view(-1)

pytorch Flatten展平相关推荐

python学习笔记之-展平函数ravel和flatten及两者的区别
ravel()和flatten()是将多维数据展平为一维数据,功能相同,区别在于一个是引用操作,一个是复制操作.ravel()展平数据后,修改后面的数据会影响前面的数据,而flatten()展平数据后 ...
Python计算两个numpy数组的交集（Intersection）实战：两个输入数组的交集并排序、获取交集元素及其索引、如果输入数组不是一维的，它们将被展平(flatten)，然后计算交集
Python计算两个numpy数组的交集(Intersection)实战:两个输入数组的交集并排序.获取交集元素及其索引.如果输入数组不是一维的,它们将被展平(flatten),然后计算交集目录
展平数组（flatten、ravel）--numpy
1. flatten()函数 1.1 函数功能返回将输入数组展平成一维的副本数组,生成的新数组是原数组的深层拷贝(copy),两个数组相互独立,对新数组的修改不会影响原数组 1.2 函数语法 nda ...
remix使用_使用Remix展平合同和调试
remix使用 Smart contracts on the Ethereum main-net use real money, so building error-free smart contra ...
Python/numpy之ravel()多维数据展平函数
Python/numpy之ravel()多维数据展平函数可参考官方文档numpy.ravel或末尾摘抄内容 ravel()将多维数据展平为一维数据,可以选择不同的数据索引方式(见文档参数四个可选值) ...
深度学习笔记（4）——维度操作（2）展平、·拼接、拆分
深度学习笔记(4)--维度操作(2)展平文章目录深度学习笔记(4)--维度操作(2)展平前言一. flatten() 二.view() 总结前言上一篇文章记录了维度操作的增加维度和维度压缩 ...
sql　数组三　展平数组
展平数组如需将 ARRAY 转换为一组行(即执行展平操作),请使用 UNNEST 运算符.UNNEST 获取一个 ARRAY,然后返回一个表,ARRAY 中的每个元素均占该表的一行. 由于 UNNE ...
flatMap() ：对每个元素执行映射函数并将结果展平
JavaScript Array flatMap() 方法介绍 flat() 方法创建一个新数组,其中连接了子数组的元素. map() 方法创建一个新数组,其元素是映射函数的结果. flatMap() ...
Maya角色UV展平技巧笔记
Maya角色UV展平技巧笔记一展平uv注意事项二展平uv步骤 1 头部uv展开 2 手臂uv展开 3 手部uv展开 4 身体uv展开 4 腿部uv展开 5 脚部uv展开三排布uv壳参考 ...

pytorch Flatten展平

pytorch Flatten展平相关推荐

最新文章

热门文章