def forward(self, xs):
    xp = cuda.get_array_module(*xs)
    return (xp.concatenate(xs, axis=self.axis),)
def forward(self, xs):
    xp = cuda.get_array_module(*xs)
    return (xp.concatenate(xs, axis=self.axis),)