求助PyTorch - pytorch吧

level 2

yepppლ 楼主

大佬们，怎么用pytorch写这个由前面所有的层过渡到7列分支这个语句；我看caffe用的是slice层，但pytorch好像没有对应的层，我是在forward函数里用了torch.chunk函数来写的，但是好像不对，图一是我写的应该不对，图二是模型图（就是怎么从红色框到蓝色框）

2022年07月10日 08点07分 1