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

Pytorch的入门

  • 1: 安装pip 包, 指定阿里云的镜像

    • pip install -i https://mirrors.aliyun.com/pypi/simple torch
  • 2: 计算机中的随机数都是伪随机数

    • 需要设定随机种子.
  • 3: 张量运算

    • data.add()
    • dada.sub()
    • data.mul()
    • data.div()
    • data.neg()
    • 都是矩阵中的每一个元素都要做对应的计算.
  • 4: 阿达母积 & 点积 (矩阵乘法)

  • data1 = [[1, 2]
             [3, 4]]
    data2 = [[5, 6]
             [7, 8]]
    
    data = torch.mul(data1, data2)
    等效于
    data = data1 * data2
    
    print(data)
    
    [[5, 12]
     [21, 32]]
    
    ----------------------------
    
    [[19, 22]
     [43, 50]]
    
  • 矩阵的广播机制

  • torch.matmul(torch.randn(3, 4, 5), torch.randn(5, 4)).shape
    
    广播之前: [3,4,5] --- [5,4]
    广播之后: [3,4,5] --- [3,5,4]
    矩阵乘法: [3,4,4]
    
    [5,4]  ---  [3,4,5]
    [3,5,4] --- [3,4,5]
    [3,5,5]
    
  • torch.cuda.is_available() 判断服务器, 电脑是否支持cuda (GPU)


http://www.kler.cn/a/585337.html

相关文章:

  • Java 8 + Tomcat 9.0.102 的稳定环境搭建方案,适用于生产环境
  • 使用curl随机间隔访问URL-使用curl每秒访问一次URL-nginx
  • Vue配置和安装教程(2025最新)
  • CGI程序处理每一帧VDEC视频数据并输出到HTML页面
  • 【Unity】TextMesh Pro显示中文部分字体异常
  • Cascadeur-3D关键帧动画软件
  • Redis--zset类型
  • 信号处理抽取多项滤波的数学推导与仿真
  • 警惕!Ollama大模型工具的安全风险及应对策略
  • Webpack 和 Vite 的主要区别
  • C# net deepseek RAG AI开发 全流程 介绍
  • flinkOracleCdc源码介绍
  • Python 与 sklearn 库:轻松构建 KNN 算法双版本
  • 如何撰写一份清晰专业的软件功能测试报告
  • Vue项目搜索引擎优化(SEO)终极指南:从原理到实战
  • JVM 垃圾回收器的选择
  • 海量数据查询加速:Presto、Trino、Apache Arrow
  • 在Vue3中集成XGPlayer视频播放器的完整指南
  • Unity打包Android平台调用sherpa-onnx
  • 【Devops】DevOps and CI/CD Pipelines