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

torch函数合集

torch.tensor()

  • 原型:torch.tensor(data, dtype=None, device=None, requires_grad=False)
  • 功能:其中data可以是:list,tuple,NumPy,ndarray等其他类型,torch.tensor会从data中的数据部分做拷贝(而不是直接引用),根据原始数据类型生成相应类型的torch.Tensor。
  • 例子:在这里插入图片描述

torch.normal()

  • 原型:normal(mean, std, *, generator=None, out=None)
  • 功能:该函数返回从单独的正态分布中提取的随机数的张量,该正态分布的均值是mean,标准差是std。
  • 例子:
    • 从一个标准正态分布N~(0,1),提取一个2x2的矩阵:torch.normal(mean=0,std=1.,size=(2,2))
      在这里插入图片描述

torch.matmul()

  • 原型:torch.matmul(input, other)
  • 功能:计算两个张量input和other的矩阵乘积
  • 例子:
    • 一维相乘:点积运算
      在这里插入图片描述 - 二维相乘:矩阵运算
      在这里插入图片描述

reshape(-1, 1), reshape(1, -1)

  • 功能:reshape(行,列)可以根据指定的数值将数据转换为特定的行数和列数
  • reshape(-1, 1):生成一列,行数由系统运算
  • reshape(1, -1):生成一行,列数由系统运算
  • 例子:
    在这里插入图片描述
  • 参考:Python的reshape的用法:reshape(1,-1)、reshape(-1,1)
http://www.lryc.cn/news/25002.html

相关文章:

  • AcWing算法提高课-3.1.2信使
  • Paddle OCR Win 11下的安装和简单使用教程
  • 杂谈:数组index问题和对象key问题
  • 三天Golang快速入门—Slice切片
  • 腾讯会议演示者视图/演讲者视图
  • 【C++】类与对象(一)
  • JavaScript基本语法
  • OpenCV4.x图像处理实例-道路车辆检测(基于背景消减法)
  • pwnlab通关流程
  • 面向过程与面向对象的区别与联系
  • 主机状态(查看资源占用情况、查看网络占用情况)
  • 代码随想录算法训练营第四十一天 | 01背包问题-二维数组滚动数组,416. 分割等和子集
  • VMware NSX 4.1 发布 - 网络安全虚拟化平台
  • 计算理论 复杂度预备知识
  • 二叉树——二叉搜索树中的插入操作
  • C# if break,if continue,if return的区别和使用
  • 力扣-第二高的薪水
  • I - 太阳轰炸(组合数学Cnk n固定)
  • centos安装gitlab
  • 【洛谷 P1093】[NOIP2007 普及组] 奖学金 题解(结构体排序)
  • 【Hello Linux】进程优先级和环境变量
  • 日期:Date,SimpleDateFormat常见API以及包装类
  • 嵌入式之ubuntu终端操作与shell常用命令详解
  • 【Shell学习笔记】6.Shell 流程控制
  • 27k入职阿里测开岗那天,我哭了,这5个月付出的一切总算没有白费~
  • 服务端开发之Java备战秋招面试篇5
  • 有趣的 Kotlin 0x11: joinToString,你真的了解嘛?
  • 代码随想录算法训练营day46 | 动态规划之背包问题 139.单词拆分
  • DPDK中的无锁共享数据结构
  • 【使用两个栈实现队列】