python类参数定义及数据扩展方式unsqueeze/expand
作者：徽先生 发布时间：2023-08-07 21:56:42
标签：Python,unsqueeze,expand
类的参数定义
将conda环境设置为ai，conda activate ai
这个文件的由来：
由于在yolov1的pytorch实现的损失函数中，看到继承了nn.Module，并且其中两个参数不像c++那里指定类型，那么他们的类型是哪里来的
这里就是在探索这样一件事
操作逻辑：
先在类中定义了构造函数以及一个自定义函数；
构造函数定义了属性S、B，自定义函数引入两个参数，对两个参数进行调用
这里就说明参数的结构是怎么样的，取决于参数被调用了什么东西，比如这里调用了
N = box1.size(0) M = box2.size(0)说明了它是类似一个矩阵的东西，对应的box1的定义就是`torch.rand(10,4)
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable
#探究属性S，B是如何产生的，以及box1、box2是如何产生的、如何调用
class yoloLoss(nn.Module):
def __init__(self,S,B):
self.S=S
self.B=B
def compute_iot(self,box1,box2):
N = box1.size(0) #调用方式就表示了变量是什么类型，这里是一个张量，其中每个元素是一个tensor，所以是N*4的张量
M = box2.size(0)
print(M,N)
yoloLoss1 =yoloLoss(10, 11)
yoloLoss1.compute_iot(torch.rand(10,4),torch.rand(11,4))
数据扩展
探究unsqueeze以及expand的使用方法，unsqueeze可以增加一个纬度，但是维度的siz只是1而已，而expand就可以将数据进行复制，将数据变为n
# 获得一开始的初始化数值：tensor([[a1,a2,a3]])
nn1=torch.rand(1,3)
print(nn1)
# unsqueeze是解压的意思，在第i个维度上进行扩展，将其扩展为tensor([[[a1,a2,a3]]])
nn1=nn1.unsqueeze(0)
print("*"*100)
print(nn1)
#利用expand对数据进行扩展
nn1=nn1.expand(1,3,3)
print("*"*100)
print(nn1)
来源：https://blog.csdn.net/weixin_42295969/article/details/126346660
0
投稿猜你喜欢
- 前言最近遇到的几个网站在提交密码时提交的已经是密文，也就是说在网络上传输的密码是密文，这样提升了密码在网络传输中的安全性。后端语言加解密已
- 在ie7发布之前，Dean的addEvent/removeEvent可以称的上是完美了。IE7发布后，引入新的内存泄漏(这个我不是很确定，忘
- 我们先看一个简单的例子：<input type="text" onblur="alert(this.va
- 定义和用法strftime() 函数根据区域设置格式化本地时间／日期。语法strftime(format,timestamp)参数 描述 f
据国外媒体报道，相较于IE8浏览器，微软最新一代浏览器IE9的最大改进就是硬件加速HTML5。微软承诺，通过利用IE9中的硬件加速功能，开发- 如果说哪个开源程序不需要介绍大家就认识，那一定是phpMyAdmin，一款流行的MySQL数据库的Web管理界面。MySQL是全球最流行的W
- 本文试图从iPhone的角度结合一些iPhone平台项目的设计经验提炼出iPhone平台的一些优秀设计思路，以供大家在做移动互联网设备设计时
- 本文是小编针对js保留两位小数这个大家经常遇到的经典问题整理了在各种情况下的函数写法以及遇到问题的分析，以下是全部内容：一、我们首先从经典的
- DreamWeaver 4的到来让大家兴奋吧？但是大家一定为DreamWeaver4里面的字体、文字大
- 简介如果你经常网上冲浪，这样参差不齐的多栏布局，是不是很眼熟啊？类似的布局，似乎一夜之间出现在国内外大大小小的网站上，比如 Pinteres
- 一、垃圾还是经典网页技术更新很快，一个网站的界面设计寿命仅仅2-3年而已。不管是垃圾还是精品，都没有所谓的经典。经典只存在于是哪个首次成功创
- 全局变量是魔鬼。在YUI中，我们仅用两个全局变量：YAHOO和YAHOO_config。YUI的一切都是使用YAHOO对象级的成员或这个成员
- 在使用SQL Server 的过程,中由于经常需要从多个不同地点将数据集中起来或向多个地点复制数据,所以数据的导出,导入是极为常见的操作.我
- 在做网站产品展示页面时，一般会用到缩略图，好处当然是直观醒目让人一目了然。点击进入然后看到大图及具体的介绍。但是缩略图在实现上带来了两个问题
- 在《JavaScript语言精粹》的第72页有这样一段：用正则表达式字面量创建的RegExp对象来共享同一个单实例：function mak
- 在广大网友心目中，他们就是中国互联网搜索领域的三驾马车。无论这三家搜索巨头承不承认，在网友眼中总会来将他们进行对比比较。当然，更多时候的比较
- Adobe AIR 在我们国庆节的时候发布了beta2版。 同时发布的还有： FLEX 3.0 beta2 官方说这一版本已经基本上接近正式
- 前言使用np.savetxt()方法可以将数据保存为txt文件或者是csv文件。1 np.savetxt()存储txt文件1-1 基础参数n
- SQL Server数据库日志清除的两个方法：方法一一般情况下，SQL数据库的收缩并不能很大程度上减小数据库大小，其主要作用是收缩日志大小，
- 定义本地站点在Dreamweaver中制作网站，您必须定义一个本地站点，它是您的计算机上任意位置的一个
