def METHOD_NAME(self, x):
    x = lbann.Reshape(x, dims=[-1, self.input_dim])
    for i, dim in enumerate(self.hidden_dims):
        x = lbann.ChannelwiseFullyConnected(
            x,
            weights=self.weights[i],
            output_channel_dims=dim,
            bias=False,
        )
        x = lbann.Relu(x)
    x = lbann.ChannelwiseFullyConnected(
        x,
        weights=self.weights[-1],
        output_channel_dims=self.output_dim,
        bias=False,
    )
    return x