HybridBlock of concat layer

concat.py

from mxnet.gluon import HybridBlock

class Concat(HybridBlock):
  """Concat operation for multiple inputs."""
  def __init__(self, dim=1, **kwargs):
    super(Concat, self).__init__(**kwargs)
    self._kwargs = {'dim': dim}
    
  def hybrid_forward(self, F, *args):
    return F.concat(*args, name='fwd', **self._kwargs)
  
  def __repr__(self):
    s = '{name}(dim={dim})'
    return s.format(name=self.__class__.__name__, **self._kwargs)

How to use this block