当前位置:网站首页>pytorch总结学习系列-广播机制
pytorch总结学习系列-广播机制
2022-06-29 08:31:00 【TJMtaotao】
前⾯我们看到如何对两个形状相同的 Tensor 做按元素运算。当对两个形状不同的 Tensor 按元素运算
时,可能会触发⼴广播(broadcasting)机制:先适当复制元素使这两个 Tensor 形状相同后再按元素
运算。例如
x = torch.arange(1, 3).view(1, 2)
print(x)
y = torch.arange(1, 4).view(3, 1)
print(y)
print(x + y)
tensor([[1, 2]])
tensor([[1],
[2],
[3]])
tensor([[2, 3],
[3, 4],
[4, 5]])
其对应的操作就是x进行了行的两次复制
y进行了列的一次复制
x=[[1,2],
[1,2],
[1,2]]
y=[[1,1],
[2,2],
[3,3]]
边栏推荐
猜你喜欢
随机推荐
微信小程序自定义多项选择器
微信小程序最新canvas2d手写签名
Mysql使用union all统计多张表组合总数,并分别统计各表数量
Mqtt second session -- emqx high availability cluster implementation
MYSQL行转列例子
AugFPN:改进多尺度特征学习用于目标检测
[to.Net] C data model, from Entity Framework core to LINQ
【目标检测】|指标 A probabilistic challenge for object detection
train_on_batch保存一下loss函数变化的图像
Uni app gets the route URL of the current page
单例模式的理解
What exactly does Devops mean?
JS获取图片或base64的宽高等基本信息
uniapp微信小程序报错 TypeError: Cannot read property ‘call‘ of undefined
Open3d hidden point removal
mongoDB 持久化
Let's make a summary
微信小程序分享页面,分享到朋友圈
15 things to learn in a year of internship in famous enterprises, so you can avoid detours.
cokkie和session的区别









