torch 中的广播机制

自学AI的鲨鱼儿

已于 2023-08-15 18:26:56 修改

阅读量404

点赞数

分类专栏：小代码库文章标签： pytorch 深度学习人工智能

于 2023-08-15 18:18:59 首次发布

本文链接：https://blog.csdn.net/qq_16555103/article/details/132304666

版权

小代码库专栏收录该内容

6 篇文章 0 订阅

订阅专栏

一、元素级操作广播机制(对应位置相加）

pytorch维度不同的张量（tensor）相加规则_张量相加

二、矩阵乘法广播机制

import torch
x = torch.randn(2,3,4,5)
print(x)

w = torch.randn(2,1,5,6)  # 维度需自右对齐，后两位维度满足为矩阵乘法维度规则，其余的维度要么完全对齐相同，要么维度为1进行广播
print(w) 
print(torch.matmul(x,w))
print(torch.matmul(x[0],w[0]))
print(torch.matmul(x[1],w[1]))