        )  # shape: [num_chunks*B, C,D,H,W]
        y = super().forward(x)  # shape: [num_chunks*B, C', D', H', W']
        _, C1, D1, H1, W1 = y.shape
        y = y.reshape(num_chunks, B, C1, D1, H1, W1)
        return y
