首页 > 其他分享 >深度学习模型训练中,输入数据维度和标签数据维度调整方法

深度学习模型训练中,输入数据维度和标签数据维度调整方法

时间:2023-09-08 23:12:31浏览次数:43  
标签:inputs 标签 labels shape 维度 数据

for inputs, labels in train_loader:
        # 使用numpy的transpose函数调整维度顺序
        inputs = np.transpose(inputs, (0, 3, 1, 2)) #将原输入数据最后一个维度换到第二个维度
        inputs = inputs.to(device)
        print(inputs.shape) #调试代码用
        m = labels.shape  #hdf5文件有时候标签数据大小为[batch_siza,1,1,1],需要做如下调整,先获取标签数据大小
        n = m[0] #获取标签数据第一个维度值,也就是batch_size
        labels = torch.reshape(labels, (n,)) #将标签数据转换为一维数据,也就是将原来的[[[1]]]里面的标签1提取出来
        labels = labels.to(device)
        print(labels.shape) #调试代码用
        # 前向传播
        outputs = model(inputs)
        print(outputs.shape) #调试代码用
        loss = criterion(outputs, labels.long())

 

标签:inputs,标签,labels,shape,维度,数据
From: https://www.cnblogs.com/xjkj/p/17688722.html

相关文章

  • R语言分析糖尿病数据:多元线性模型、MANOVA、决策树、典型判别分析、HE图、Box's M检验
    全文链接:https://tecdat.cn/?p=33609原文出处:拓端数据部落公众号背景Reaven和Miller(1979)研究了145名非肥胖成年人的葡萄糖耐量和胰岛素血液化学指标之间的关系。他们使用斯坦福线性加速器中心的PRIM9系统将数据可视化为3D,并发现了一个奇特的图案,看起来像是一个有两个翼的大斑点......
  • R语言混合图形模型MGM的网络可预测性分析|附代码数据
    原文链接:http://tecdat.cn/?p=18279 最近我们被客户要求撰写关于混合图形模型MGM的研究报告,包括一些图形和统计输出。网络模型已经成为抽象复杂系统,是深入了解许多科学领域中观测变量之间的关系模式的流行方法。这些应用程序大多数集中于分析网络的结构。但是,如果不是直接观察......
  • R语言统计学DOE实验设计:用平衡不完全区组设计(BIBD)分析纸飞机飞行时间实验数据|附代码
    全文链接:http://tecdat.cn/?p=31010原文出处:拓端数据部落公众号平衡不完全区组设计(BIBD)是一个很好的研究实验设计,具有从统计的角度看各种所需的特征。最近我们被要求撰写关于BIBD的研究报告,包括一些图形和统计输出。对于一个BIBD有K个观测,重复r次实验。还有第5参数lamda,记录其......
  • R语言武汉流动人口趋势预测:灰色模型GM(1,1)、ARIMA时间序列、logistic逻辑回归模型|附代
    全文链接:http://tecdat.cn/?p=32496原文出处:拓端数据部落公众号人口流动与迁移,作为人类产生以来就存在的一种社会现象,伴随着人类文明的不断进步从未间断。人力资源是社会文明进步、人民富裕幸福、国家繁荣昌盛的核心推动力量。当前,我国经济正处于从以政府主导的投资驱动型的经......
  • SAS数据挖掘EM贷款违约预测分析:逐步Logistic逻辑回归、决策树、随机森林|附代码数据
    全文链接:http://tecdat.cn/?p=31745原文出处:拓端数据部落公众号最近我们被客户要求撰写关于贷款违约预测的研究报告,包括一些图形和统计输出。近几年来,各家商业银行陆续推出多种贷款业务,如何识别贷款违约因素已经成为各家商业银行健康有序发展贷款业务的关键。在贷款违约预测的......
  • laravel 通过事务处理两个并发请求同时更新同一条 MySQL 数据
    在处理两个并发请求同时更新同一条MySQL数据的情况时,你需要考虑并发冲突的问题,以确保数据的一致性和完整性。以下是一些解决方法:在Laravel中,你可以使用数据库事务来处理并发更新。在事务中,你可以执行一系列操作,并在操作完成后提交或回滚事务。如果两个请求同时尝试更新相同的......
  • Java学习_004 数据输入:案例2
    需求:三个和尚的身高需要手动输入,请用程序实现这三个和尚的最高身高。importjava.util.Scanner;publicclassMain{publicstaticvoidmain(String[]args){Scannersc=newScanner(System.in);intheight1=sc.nextInt();intheight2......
  • Java学习_003 数据输入
    1.数据输入1.2Scanner使用的基本步骤(1)导包importjava.util.Scanner;(2)创建对象Scannersc=newScanner(System.in);(3)接受数据inti=sc.nextInt();1.3实例importjava.util.Scanner;publicclassMain{pub......
  • HTML5元素分类和新增标签
    元素分类<!DOCTYPEhtml><htmllang="en"><head><metacharset="UTF-8"><metaname="viewport"content="width=device-width,initial-scale=1.0"><title>Document</title&g......
  • HTML5表格标签和单元格合并
    表格标签<!DOCTYPEhtml><htmllang="en"><head><metacharset="UTF-8"><metaname="viewport"content="width=,initial-scale=1.0"><title>Document</title></head&g......