MXNet HybridBlock模型并行实现咨询：基于示例代码的修改问题

作为刚入坑MXNet的新手，我最近一直在摸索模型并行技术，偶然发现了一篇超实用的文章——《MXNet模型并行简单示例》。参考这篇文章的思路，我把里面的代码改成了使用HybridBlock的版本，修改后的代码如下：

import numpy as np
import mxnet as mx
from mxnet import nd, autograd, gluon
from mxnet.gluon import HybridBlock

num_inputs = 2
num_outputs = 1
num_examples = 10000

def real_fn(x):
    return 2 * x[:, 0] - 3.4 * x[:, 1] + 4.2

x = np.random.normal(0, 1, (num_examples, num_inputs))
noise = 0.001 * np.random.normal(0, ...

内容的提问来源于stack exchange，提问作者Hugo