IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> torch.squeeze()和torch.unsqueeze()简要介绍 -> 正文阅读

[人工智能]torch.squeeze()和torch.unsqueeze()简要介绍

定义

  • torch.squeeze(input,dim)
  • torch.unsqueeze(input,dim)

用法

  • torch.squeeze():对输入数据的维度dim进行压缩,默认删除所有为1的维度
  • torch.unsqueeze():对输入数据的维度dim进行扩充,dim为必填参数

示例

torch.squeeze()

import torch

a = torch.tensor([[2],[3],[1],[0],[5],[6]])
print(torch.squeeze(a))
print(torch.squeeze(a,0))
print(torch.squeeze(a,1))

>>>tensor([2, 3, 1, 0, 5, 6])

tensor([[2],
        [3],
        [1],
        [0],
        [5],
        [6]])

tensor([2, 3, 1, 0, 5, 6])

torch.unsqueeze()

import torch

a = torch.tensor([[2],[3],[1],[0],[5],[6]])

print(torch.unsqueeze(a,0))
print(torch.unsqueeze(a,1))

>>>tensor([[[2],
            [3],
            [1],
            [0],
            [5],
            [6]]])

tensor([[[2]],

        [[3]],

        [[1]],

        [[0]],

        [[5]],

        [[6]]])
?

  人工智能 最新文章
Neo4j 实战篇(一)基于Py2Neo构建电影知识
消息传递神经网络(MPNN)内容及代码实践
OpenCV+百度云人脸识别项目及源码
实验5 卷积神经网络实验中
论文详解:Attention-based Extraction of
Tensorflow进阶
科技时代,如何保护个人隐私?
【元宇宙系列】游戏与元宇宙(下)——拉动
pytorch学习13:实现LetNet和学习nn.Module
LSTM的本质理解-正向传播
上一篇文章      下一篇文章      查看所有文章
加:2022-01-17 11:30:57  更:2022-01-17 11:33:28 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
360图书馆 购物 三丰科技 阅读网 日历 万年历 2022年5日历 -2022/5/24 5:59:28-
图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码