type
status
date
slug
summary
tags
category
icon
password

PyTorch广播机制

定义

PyTorch的tensor参数可以自动扩展其大小。一般的是小一点的会变大,来满足运算需求。

规则

满足一下情况的tensor是可以广播的。
  • 至少有一个维度
  • 两个tensor维度相等
  • 维度不等,其中一个为1
  • 维度不等,其中一个维度不存在
计算过程规则。
  • 维度不同,小维度的增加维度
  • 每个维度,计算结果取大的
  • 扩展维度是对数值进行复制

例子

输出结果
 
Python 打印表格shell脚本
Ghlerrix
Ghlerrix
山高水长,怕什么来不及,慌什么到不了。
公告
type
status
date
slug
summary
tags
category
icon
password
这是新版的博客,想要访问旧版页面请使用以下链接。