chainer.functions.squeeze(x, axis=None)[source]

ndarrayのshapeからサイズ１の次元を除去する。

 

Parameters:
  • x (Variable or numpy.ndarray or cupy.ndarray) –入力値. A (s1,s2,...,sN)(s1,s2,...,sN) -型の float 配列.
  • axis (None or int or tuple of ints) – Shapeの中から取り除く１次元のエントリのサブセット。 None が与えられた場合、全てが除去される。次元数は０から始まる。１より大きい次元の軸が選ばれた場合、エラーが発生する。
Returns:

次元サイズが１のものが除去されたVariable 。

Return type:

Variable

 

Example

  


>>> x = np.array([[[[0, 1, 2]]], [[[3, 4, 5]]]], 'f')

>>> x.shape
(2, 1, 1, 3)

>>> y = F.squeeze(x)

>>> y.shape
(2, 3)

>>> y.data

array([[ 0.,  1.,  2.],
       [ 3.,  4.,  5.]], dtype=float32)

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

>>> y.shape
(2, 1, 3)

>>> y.data

array([[[ 0.,  1.,  2.]],

       [[ 3.,  4.,  5.]]], dtype=float32)

>>> y = F.squeeze(x, axis=(1, 2))

>>> y.shape
(2, 3)

>>> y.data

array([[ 0.,  1.,  2.],
       [ 3.,  4.,  5.]], dtype=float32)