def __repr__(self):
    s = '{block_name}({input_dim} -> {output_dim}, {dtype})'
    return s.format(block_name=self.__class__.__name__, **self._kwargs)
def __repr__(self):
    s = '{block_name}({input_dim} -> {output_dim}, {dtype})'
    return s.format(block_name=self.__class__.__name__, **self._kwargs)