xiaoidngding的博客

原创 gab跳跃连接

xl = torch.randn(1, 64, 16, 16) # 输入 xl 的形状为 [B C H/2 W2]mask = torch.randn(1, 1, 16, 16) # 蒙版张量的形状为 [B 1 H/2 W/2]xh = torch.randn(1, 64, 32, 32) # 输入 xh 的形状为 [B C H W]print("输入 xh 的形状:", xh.size())print("输入 xl 的形状:", xl.size())# 创建模拟输入数据。#可以缝合在跳跃连接部分。