Class Reshape

tensorlayer/layers/shape.py:61–101

A layer that reshapes a given tensor. Parameters ---------- shape : tuple of int The output shape, see ``tf.reshape``. name : str A unique layer name. Examples -------- >>> x = tl.layers.Input([8, 4, 3], name='input') >>> y = tl.layers.Reshape(shape=

class Reshape(Layer):
    """A layer that reshapes a given tensor.

    Parameters
    ----------
    shape : tuple of int
        The output shape, see ``tf.reshape``.
    name : str
        A unique layer name.

    Examples
    --------
    >>> x = tl.layers.Input([8, 4, 3], name='input')
    >>> y = tl.layers.Reshape(shape=[-1, 12], name='reshape')(x)
    (8, 12)

    """

    def __init__(self, shape, name=None):  #'reshape'):
        super(Reshape, self).__init__(name)
        self.shape = shape

        logging.info("Reshape %s" % (self.name))

        self.build()
        self._built = True

    def __repr__(self):
        s = '{classname}('
        s += 'shape={shape},'
        s += 'name=\'{name}\''
        s += ')'
        return s.format(classname=self.__class__.__name__, **self.__dict__)

    def build(self, inputs_shape=None):
        pass

    # @tf.function
    def forward(self, inputs):
        outputs = tf.reshape(inputs, shape=self.shape, name=self.name)
        return outputs


class Transpose(Layer):