语句
[MC(self.shape) for _ in range(self.layer)]
或:
[MC(self.shape).cuda() for _ in range(self.layer)]
解释:
如创建卷积层的语句:
[MyCon(self.shape) for _ in range(self.layer)]
如果在一个对象中需要创建多个卷积层,可以在自己定义的类(如MyClass)的初始化中使用:
self.myconv = [MyCon(self.shape) for _ in range(self.layer)]
但是有时会导致数据不在同一设备问题问题
MyClass.cuda()
语句并不会将创建的一系列MyCon类放入到GPU设备中。可以使用以下语句:
[MyCon(self.shape).cuda() for _ in range(self.layer)]