当前位置：网站首页 > 技术分类 > 正文

pytorch的广播机制（python的广播运算）

ztj100 2024-11-14 19:24 11 浏览 0 评论

我们知道，两个形状形同的张量可以按照元素进行算数运算，如果他们的形状不一样，还能进行算数运算吗？

如果形状不一样也是可以的，这就是pytorch的广播机制。

但是广播机制有个前提条件，参与运算的两个张量，必须有一个维度的大小为1，且为1的维度不同。也就是说，假设是两个2维张量进行运算，一个张量的列维度大小为1，另一个张量的行维度大小为1。

满足这个条件的张量在运算时，pytorch会自动把大小为1的维度自动复制成与另一个张量同维度相同的大小。

看个例子。

import torch
a = torch.arange(3).reshape((1, 3))
b = torch.arange(2).reshape((2, 1))
a+b

构建2个2维张量，一个1行3列，一个2行1列，pytorch会先把两个张量都扩张成2行3列，然后按元素相加。

torch.arange

上一篇：相对位置编码原理写了一个例子写PyTorch 代码
下一篇：PyTorch自动求导（pytorch自动求导逻辑）

相关推荐

从IDEA开始，迈进GO语言之门（idea got）: 前言笔者在学习GO语言编程的时候，GO语言在国内还没有像JAVA/Php/Python那样普及，绕了不少的弯路，要开始入门学习一门编程语言，最好就先从选择一个好的编程语言的开发环境开始，有了这个开发环...

基于SpringBoot+MyBatis的私人影院java网上购票jsp源代码Mysql: 本项目为前几天收费帮学妹做的一个项目，JavaEEJSP项目，在工作环境中基本使用不到，但是很多学校把这个当作编程入门的项目来做，故分享出本项目供初学者参考。一、项目介绍基于SpringBoot...

基于springboot的个人服装管理系统java网上商城jsp源代码mysql: 本项目为前几天收费帮学妹做的一个项目，JavaEEJSP项目，在工作环境中基本使用不到，但是很多学校把这个当作编程入门的项目来做，故分享出本项目供初学者参考。一、项目介绍基于springboot...

基于springboot的美食网站Java食品销售jsp源代码Mysql: 本项目为前几天收费帮学妹做的一个项目，JavaEEJSP项目，在工作环境中基本使用不到，但是很多学校把这个当作编程入门的项目来做，故分享出本项目供初学者参考。一、项目介绍基于springboot...

贸易管理进销存springboot云管货管账分析java jsp源代码mysql: 本项目为前几天收费帮学妹做的一个项目，JavaEEJSP项目，在工作环境中基本使用不到，但是很多学校把这个当作编程入门的项目来做，故分享出本项目供初学者参考。一、项目描述贸易管理进销存spring...

SpringBoot+VUE员工信息管理系统Java人员管理jsp源代码Mysql: 本项目为前几天收费帮学妹做的一个项目，JavaEEJSP项目，在工作环境中基本使用不到，但是很多学校把这个当作编程入门的项目来做，故分享出本项目供初学者参考。一、项目介绍SpringBoot+V...

目前见过最牛的一个SpringBoot商城项目(附源码)还有人没用过吗: 帮粉丝找了一个基于SpringBoot的天猫商城项目，快速部署运行，所用技术：MySQL，Druid，Log4j2，Maven，Echarts，Bootstrap...免费给大家分享出来前台演示...

SpringBoot+Mysql实现的手机商城附带源码演示导入视频: 今天为大家带来的是基于SpringBoot+JPA+Thymeleaf框架的手机商城管理系统，商城系统分为前台和后台、前台用的是Bootstrap框架后台用的是SpringBoot+JPA都是现在主...

全网首发!马士兵内部共享—1658页《Java面试突击核心讲》: 又是一年一度的“金九银十”秋招大热门，为助力广大程序员朋友“面试造火箭”，小编今天给大家分享的便是这份马士兵内部的面试神技——1658页《Java面试突击核心讲》！...

SpringBoot数据库操作的应用（springboot与数据库交互）: 1.JDBC+HikariDataSource...

SpringBoot 整合 Flink 实时同步 MySQL: 1、需求在Flink发布SpringBoot打包的jar包能够实时同步MySQL表，做到原表进行新增、修改、删除的时候目标表都能对应同步。...

SpringBoot + Mybatis + Shiro + mysql + redis智能平台源码分享: 后端技术栈基于SpringBoot+Mybatis+Shiro+mysql+redis构建的智慧云智能教育平台基于数据驱动视图的理念封装element-ui，即使没有vue的使...

Springboot+Mysql舞蹈课程在线预约系统源码附带视频运行教程: 今天发布的是由【猿来入此】的优秀学员独立做的一个基于springboot脚手架的Springboot+Mysql舞蹈课程在线预约系统，系统项目源代码在【猿来入此】获取！https://www.yuan...

SpringBoot+Mysql在线众筹系统源码+讲解视频+开发文档(参考论文: 今天发布的是由【猿来入此】的优秀学员独立做的一个基于springboot脚手架的在线众筹管理系统，主要实现了普通用户在线参与众筹基本操作流程的全部功能，系统分普通用户、超级管理员等角色，除基础脚手架外...

Docker一键部署 SpringBoot 应用的方法，贼快贼好用: 这两天发现个Gradle插件，支持一键打包、推送Docker镜像。今天我们来讲讲这个插件，希望对大家有所帮助！GradleDockerPlugin简介...

取消回复欢迎你发表评论:

一周热门

最近发表

首页
收录
顶部