Class Stack

tensorlayer/layers/stack.py:16–64

The :class:`Stack` class is a layer for stacking a list of rank-R tensors into one rank-(R+1) tensor, see `tf.stack() `__. Parameters ---------- axis : int New dimension along which to stack. name : str A uniq

class Stack(Layer):
    """
    The :class:`Stack` class is a layer for stacking a list of rank-R tensors into one rank-(R+1) tensor, see `tf.stack() <https://www.tensorflow.org/api_docs/python/tf/stack>`__.

    Parameters
    ----------
    axis : int
        New dimension along which to stack.
    name : str
        A unique layer name.

    Examples
    ---------
    >>> import tensorflow as tf
    >>> import tensorlayer as tl
    >>> ni = tl.layers.Input([None, 784], name='input')
    >>> net1 = tl.layers.Dense(10, name='dense1')(ni)
    >>> net2 = tl.layers.Dense(10, name='dense2')(ni)
    >>> net3 = tl.layers.Dense(10, name='dense3')(ni)
    >>> net = tl.layers.Stack(axis=1, name='stack')([net1, net2, net3])
    (?, 3, 10)

    """

    def __init__(
        self,
        axis=1,
        name=None,  #'stack',
    ):
        super().__init__(name)
        self.axis = axis

        self.build(None)
        self._built = True
        logging.info("Stack %s: axis: %d" % (self.name, self.axis))

    def __repr__(self):
        s = '{classname}(axis={axis}'
        if self.name is not None:
            s += ', name=\'{name}\''
        s += ')'
        return s.format(classname=self.__class__.__name__, **self.__dict__)

    def build(self, inputs_shape):
        pass

    def forward(self, inputs):
        outputs = tf.stack(inputs, axis=self.axis, name=self.name)
        return outputs


class UnStack(Layer):