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

代码解读:y.view(y.size(0), -1)---tensor张量第一维保持不变,其余维度展平

y.view(y.size(0), -1)代码解读:

用于改变PyTorch张量(tensor)y的形状的。
y.size(0)返回y的第一维的大小。
-1表示让PyTorch自动计算该维度的大小,以确保新的张量与原始张量有相同的元素数量。
功能:将y的第一维保持不变,然后将其余的维度展平。

#!/usr/bin/env python3
import torchx = torch.randn((3, 1, 2))
print(x)
print(x.size(0))
print(x.view(x.size(0), -1))y = torch.randn((3, 2, 2))
print(y)
print(y.size(0))
print(y.view(y.size(0), -1))z = torch.randn((2,3))
print(z)
print(z.size(0))
print(z.view(z.size(0), -1))
# x输出
tensor([[[ 1.3927,  0.8704]],[[-2.2128,  0.1680]],[[ 1.3459,  1.6155]]])
3
tensor([[ 1.3927,  0.8704],[-2.2128,  0.1680],[ 1.3459,  1.6155]])
# y输出
tensor([[[-1.0421,  0.2478],[-1.5509,  1.5805]],[[ 0.6882,  0.4131],[-0.0034,  0.1004]],[[ 0.2090, -0.7472],[ 1.2551,  0.4388]]])
3
tensor([[-1.0421,  0.2478, -1.5509,  1.5805],[ 0.6882,  0.4131, -0.0034,  0.1004],[ 0.2090, -0.7472,  1.2551,  0.4388]])
# z输出
tensor([[-0.1616, -0.6525, -1.9303],[-1.4292, -2.0358,  0.2268]])
2
tensor([[-0.1616, -0.6525, -1.9303],[-1.4292, -2.0358,  0.2268]])
http://www.lryc.cn/news/170998.html

相关文章:

  • 必示科技赋能广发证券运维数字化实践案例,入选信通院《中国AIOps现状调查报告(2023)》
  • 特斯拉Dojo超算:AI训练平台的自动驾驶与通用人工智能之关键
  • Linux中的一些常用命令
  • VRTK4⭐二.VRTK4的项目基础配置
  • word-doc和docx区别
  • 深度学习-偏导数复习
  • linux之jq命令
  • nginx知识点详解:反向代理+负载均衡+动静分离+高可用集群
  • powerDesigner 的基本使用
  • Java下打印一个等腰三角型
  • HR的职业规划
  • avi怎么转换成视频?
  • 爬虫数据存储:技术、策略与实践(一)
  • 【音视频】ffplay解析-音视频同步
  • 虚拟列表 - Vue3实现一个可动态改变高度的虚拟滚动列表
  • PyTorch实战:实现Cifar10彩色图片分类
  • Vue模板语法(下)
  • uniapp掉完接口后刷新当前页面方法
  • linux安装redis超级详细教程
  • 2023-09-20 事业-代号z-个人品牌-数据库内核专家-分析
  • UVA-1343 旋转游戏 题解答案代码 算法竞赛入门经典第二版
  • 【运维篇】二、配置文件与多环境控制
  • 【WFA】 VHT-5.2.27 Pre-requisite throughput lower than expected
  • Pytorch史上最全torch全版本离线文件下载地址大全(9月最新)
  • CentOS服务器利用docker搭建中间件命令集合
  • Flask狼书笔记 | 09_图片社交网站 - 长文
  • 【链表】K 个一组翻转链表-力扣 25 题
  • jdk17新特性
  • 爬虫项目(四):抓取网页所有图片
  • 短剧推广和小说推文在哪里授权介绍