pytorch ----【输入张量.data.size()/输入张量.size()】的使用

embedded/2024/9/29 11:14:49/
import torch
test_data = torch.randn(1, 12, 32, 32)
batchsize, num_channels, height, width = test_data.data.size()

在提供的代码中,test_data 是一个形状为 (1, 12, 32, 32) 的随机张量,表示一个批次(batch)中有 1 张图像,每张图像有 12 个通道,图像的高度和宽度均为 32 像素。

注意事项

  • test_data.size() 返回一个包含张量各个维度大小的元组,可以直接解包到多个变量中。
  • 确保在调用 .size() 时,使用的是 size() 而不是 data.size(),后者在新的 PyTorch 版本中已不推荐使用。

 打印内容:

print(f"Batch Size: {batchsize}, Channels: {num_channels}, Height: {height}, Width: {width}")
#Batch Size: 1, Channels: 12, Height: 32, Width: 32


http://www.ppmy.cn/embedded/119313.html

相关文章

Reis数据库及key的操作命令汇总

管理命令 命令含义BGREWRITEAOF异步执行一个 AOF(AppendOnly File) 文件重写操作BGSAVE在后台异步保存当前数据库的数据到磁盘CLIENT关闭客户端连接CLIENT LIST获取连接到服务器的客户端连接列表CLIENT GETNAME获取连接的名称CLIENT PAUSE在指定时间内终…

npm 源切换以及添加 使用工具 nrm 使用方法

nrm (npm Registry Manager) 是一个用于管理和切换 npm 源的命令行工具,可以帮助开发者更方便地在不同的 npm 源之间切换,特别是对于需要频繁切换源的用户来说非常有用。以下是 nrm 的基本使用方法: 1. 安装 nrm 首先,你需要全局…

【Qt】Qt中的窗口坐标 信号与槽

Qt中的窗口坐标 && 信号与槽 1. Qt中的窗口坐标2. 信号与槽的概述3. 信号和槽的使用3.1 connect函数的使用3.2 查看内置信号和槽3.2 connect的参数类型不匹配问题 4. 自定义信号 && 自定义槽4.1 自定义槽4.2 自定义信号 5. 带参数的信号和槽6. 信号与槽的关联方…

教师工作量|基于springBoot的教师工作量管理系统设计与实现(附项目源码+论文+数据库)

私信或留言即免费送开题报告和任务书(可指定任意题目) 目录 一、摘要 二、相关技术 三、系统设计 四、数据库设计 五、核心代码 六、论文参考 七、源码获取 一、摘要 传统信息的管理大部分依赖于管理人员的手工登记与管理&#xff…

记录linux环境下搭建本地MQTT服务器实现mqtt的ssl加密通讯

1、ubuntu安装mosquitto sudo apt-get update//安装服务端 sudo apt-get install mosquitto//安装客户端 sudo apt-get install mosquitto-clients 2、安装openssl 3、mqtts/tls加密传输 mosquitto原生支持了TLS加密,TLS(传输层安全)是SSL&…

基于python+spark的外卖餐饮数据分析系统设计与实现(含论文)-Spark毕业设计选题推荐

博主介绍: 大家好,本人精通Java、Python、C#、C、C编程语言,同时也熟练掌握微信小程序、Php和Android等技术,能够为大家提供全方位的技术支持和交流。 我有丰富的成品Java、Python、C#毕设项目经验,能够为学生提供各类…

Word样式的同步与重置

有时候我们需要修改Word中的样式,实现排版的个性化。 如何同步样式到其他电脑上? Word中的样式是由Normal.dotm文件控制的,对样式所有的设置和修改,都会保存到这个问题件中,所以我们只需要在设置好样式以后&#xff…

JSP(Java Server Pages)基础使用二

简单练习在jsp页面上输出出乘法口诀表 既然大家都是来看这种代码的人了&#xff0c;那么这种输出乘法口诀表的这种简单算法肯定是难不住大家了&#xff0c;所以这次主要是来说jsp的使用格式问题。 <%--Created by IntelliJ IDEA.User: ***Date: 2024/7/18Time: 11:26To ch…