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

ops/2024/10/20 5:41:30/
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/ops/118068.html

相关文章

数据治理003-数据域

数据仓库是面向主题(数据综合、归类并进行分析利用的抽象)的应用。 数据仓库模型设计除横向的分层外,通常也需要根据业务情况进行纵向划分数据域。数据域是联系较为紧密的数据主题的集合,通常是根据业务类别、数据来源、数据用途…

some 牛课题

D-Cidoai的字符集合_牛客练习赛128 (nowcoder.com) #include "bits/stdc.h" #define int long long using namespace std; int fa[100000000]; int sz[100000000]; int find(int n) {if(n!fa[n])fa[n]find(fa[n]);return fa[n]; } map<string,int>mp; signed …

ArcGIS Desktop使用入门(三)常用工具条——拓扑(上篇:地图拓扑)

系列文章目录 ArcGIS Desktop使用入门&#xff08;一&#xff09;软件初认识 ArcGIS Desktop使用入门&#xff08;二&#xff09;常用工具条——标准工具 ArcGIS Desktop使用入门&#xff08;二&#xff09;常用工具条——编辑器 ArcGIS Desktop使用入门&#xff08;二&#x…

H.264与H.265

在使用海康平台做涉嫌头配置时&#xff0c;有H.264和H.265。大家可能搞不清楚两者的区别&#xff0c;下边我们来看一下H.264和H.265的区别。 H.264与H.265的区别 压缩效率&#xff1a; H.264&#xff1a;由于算法优化&#xff0c;H.264可以在低于1Mbps的速度下实现标清数字图像…

【递归】8. leetcode 671 二叉树中第二小的节点

题目描述 题目链接&#xff1a;二叉树中第二小的节点 2 解答思路 注意这句话&#xff1a;该节点的值等于两个子节点中较小的一个 二叉树的根节点的值是整棵树中最小的值 本道题所要求的是二叉树中第二小的节点。因为根节点是最小的节点&#xff0c;那么我们只需要找到第一…

Android面试必备知识:Android络访问框架对比(特点、使用高级技巧、使用场景)

目录 1 综合对比 2 Android网络访问框架基础理论 2.1 原生HttpUrlConnection 2.1.1 基本原理 2.1.2 优缺点分析 2.2 Android-async-http框架 2.2.1 特点与优势 2.2.2 使用场景 2.3 Volley框架 2.3.1 架构设计 2.3.2 性能特点 2.3.3 适用场景 2.4 Retrofit框架 2.4…

node.js + vue

安装Node.js和vue-cli 参考https://mp.csdn.net/mp_blog/creation/editor/142427043 生成Vue demo和编译Vue来生成静态文件&#xff08;包括index.html&#xff0c;js文件等等&#xff09; 参考https://mp.csdn.net/mp_blog/creation/editor/142427043 生成一个nodejs serve…

别再使用[]来获取字典的值了,来尝试一下这些方法

字典 在Python中&#xff0c;字典&#xff08;Dictionary&#xff09;是一种非常灵活的数据结构&#xff0c;用于存储键值对&#xff08;key-value pairs&#xff09;。每个键都是唯一的&#xff0c;并且与某个值相关联。字典是Python中处理映射关系&#xff08;即一个键对应一…