chainer.functions.expand_dims(x, axis)[source]

入力値の次元ををコピーせずに拡張する。

 

Parameters:
  • x (Variable or numpy.ndarray or cupy.ndarray) –入力値
  • axis (int) – 新しい軸が挿入される位置。 この axis パラメータは \(-ndim - 1 \leq axis \leq ndim\). (ndim は入力値の次元). \(axis < 0\)のとき、結果は\(ndim + 1 - |axis|\)と同様になる、
Returns:

Variable that holds a expanded input. The ndim of output is one grater than that of x.

Return type:

Variable

 

Example

  


>>> x = np.array([1, 2, 3])

>>> x.shape
(3,)

>>> y = F.expand_dims(x, axis=0)

>>> y.shape
(1, 3)

>>> y.data

array([[1, 2, 3]])

>>> y = F.expand_dims(x, axis=1)

>>> y.shape
(3, 1)

>>> y.data
array([[1],
       [2],
       [3]])

>>> y = F.expand_dims(x, axis=-2)

>>> y.shape
(1, 3)

>>> y.data

array([[1, 2, 3]])