pytorch中有没有类似目前版本的torch中的concattable 或者 paralleltable 的表?

pytorch中有没有类似目前版本的torch中的concattable 或者 paralleltable东西?
已邀请:
匿名用户

匿名用户

赞同来自:

目前没有,但是你可以使用autograd很容易地实现。
例如,连接表:
class MyModel(nn.Module):
def __init__(self):
self.submodule1 = MySubmodule1()
self.submodule2 = MySubmodule2()

def forward(self, x):
# this is equivalent to ConcatTable containing both submodules,
# but is much more readable
concat = [self.submodule1(x), self.submodule2(x)]
# do some other processing on concat...

要回复问题请先登录注册