当前位置: 首页 > article >正文

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

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.kler.cn/a/323492.html

相关文章:

  • LSTM(长短期记忆网络)详解
  • springboot004基于springboot004网页时装购物系统(源码+包运行+LW+技术指导)
  • 使用Python实现深度学习模型:智能食品配送优化
  • [Codesys]常用功能块应用分享-BMOV功能块功能介绍及其使用实例说明
  • 【Framework系列】UnityEditor调用外部程序详解
  • SpringBoot整合Mybatis-Plus实践汇总
  • 基于MATLAB的虫害检测系统
  • Java实现找色和找图功能
  • 每天一道面试题(20):锁的发生原因和避免措施
  • C++ | 定长内存池 | 对象池
  • 【C语言】动态内存管理:malloc、calloc、realloc、free
  • 每天一道面试题(19):Spring Boot 中自动装配机制的原理
  • IIS开启后https访问出错net::ERR_CERT_INVALID
  • EasyExcel使用介绍
  • 【个人笔记】数据一致性的解决方案
  • 10.C++程序中的循环语句
  • RS485ESD-Enhanced, Fail-safe, Slew-Rate-limited RS-485/RS-422 Transceivers
  • 基于Hive和Hadoop的白酒分析系统
  • 信号处理: Block Pending Handler 与 SIGKILL/SIGSTOP 实验
  • 开关电源要做哪些测试?
  • Docker精讲:基本安装,简单命令及核心概念
  • ①无需编程 独立通道 Modbus主站EtherNet/IP转ModbusRTU/ASCII工业EIP网关串口服务器
  • 染色算法的简单概述
  • altera FPGA下载失败
  • MySQL之基础篇
  • 【bug fixed】hexo d的时候Spawn failed