# 广播机制

# 当对两个形状不同的Tensor按元素运算时,
# 可能会触发广播(broadcasting)机制:先适当复制元素使这两个Tensor形状相同后再按元素运算
import torch

x = torch.arange(1, 3).view(1, 2)
print(x, 'x')
y = torch.arange(1, 4).view(3, 1)
print(y, 'y')
print(x + y, 'x+y')

在这里插入图片描述
这里相当于
]

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐