chainer.functions.split_axis(x, indices_or_sections, axis, force_tuple=True)[source]

与えられたVariablesを指定された軸で分割する。

 

Parameters:
  • x (Variable or numpy.ndarray or cupy.ndarray) – 分割されるVariable。
  • indices_or_sections (int or 1-D array) –　この引数が整数Nの場合、配列は指定された軸で均等にN分割される。これが１次元のソートされた整数配列である場合、配列が分割される位置を示している。
  • axis (int) –入力値を分割する軸。
  • force_tuple (bool) –  True を指定した場合(デフォルトはTrue)　、このメソッドは出力値の数が１個の場合でもタプルを返します。False の場合は、 出力値の数が１個の場合Variableを返す。
Returns:

出力値の数が１以上のとき、 Variable  オブジェクトのタプルを返し、それ以外はVariable を返す。 force_tuple がTrueの場合、出力値の数にかかわらず、戻り値は常にタプル。

Return type:

tuple or Variable

Note

この関数では、少なくとも出力値のうちのひとつが０サイズに分割されようとするとき ValueError が発生します。 (i.e.shapeがゼロであるような、 axis-番目の値)。