SpaceToDepth

バージョン名: SpaceToDepth-1

カテゴリー: データ移動

簡単な説明: SpaceToDepth 操作は、入力テンソルの空間次元のデータを出力テンソルの深さ次元に再配置します。

詳細な説明:

SpaceToDepth 操作は、形状 [N, C, D1, D2, ..., DK] を持つ入力テンソルの要素を、入力空間次元 D1, D2, ..., DK から新しい深さ次元の値が移動される出力テンソルに並べ替えます。

この操作は、形状 [N, C, D1, D2, ..., DK]K 空間次元を持つ入力テンソル dataY 出力テンソルに変換する次の変換と等価です。mode = blocks_first の場合:

x' = reshape(data, [N, C, D1 / block_size, block_size, D2 / block_size, block_size, ... , DK / block_size, block_size])

x'' = transpose(x',  [0,  3, 5, ..., K + (K + 1), 1,  2, 4, ..., K + K])

y = reshape(x'', [N, C * (block_size ^ K), D1 / block_size, D2 / block_size, ... , DK / block_size])

mode = depth_first の場合:

x' = reshape(data, [N, C, D1 / block_size, block_size, D2 / block_size, block_size, ..., DK / block_size, block_size])

x'' = transpose(x', [0,  1, 3, 5, ..., K + (K + 1),  2, 4, ..., K + K])

y = reshape(x'', [N, C * (block_size ^ K), D1 / block_size, D2 / block_size, ..., DK / block_size])

属性:

  • block_size

    • 説明: 移動する値ブロックのサイズを指定します。空間次元は block_size で均等に分割する必要があります。

    • 値の範囲: 正の整数

    • タイプ: int

    • デフォルト値: 1

    • 必須: いいえ

  • mode

    • 説明: 出力の深さ次元をブロック座標と古い深さの次元から収集する方法を指定します。

    • 値の範囲:

      • blocks_first: 出力の深さは [block_size, ..., block_size,  C] から収集されます

      • depth_first: 出力の深さは [C, block_size, ..., block_size] から収集されます

    • タイプ: string

    • 必須: はい

入力:

  • 1: data - rank >= 3 の T タイプの入力テンソル。必須。

出力:

  • 1: タイプ T および形状 [N, C * (block_size ^ K), D1 / block_size, D2 / block_size, ..., DK / block_size] の並べ替えられたテンソル。

タイプ:

  • T: サポートされるタイプ。

例:

 <layer type="SpaceToDepth" ...>
     <data block_size="2" mode="blocks_first"/>
     <input>
         <port id="0">
             <dim>5</dim>
             <dim>7</dim>
             <dim>4</dim>
             <dim>6</dim>
         </port>
     </input>
     <output>
         <port id="1">
             <dim>5</dim>    <!-- data.shape[0] -->
             <dim>28</dim>   <!-- data.shape[1] * (block_size ^ 2) -->
             <dim>2</dim>    <!-- data.shape[2] / block_size -->
             <dim>3</dim>    <!-- data.shape[3] / block_size -->
         </port>
     </output>
 </layer>