DepthToSpace¶
バージョン名: DepthToSpace-1
カテゴリー: データ移動
簡単な説明: DepthToSpace 操作は、入力テンソルの深さ次元のデータを出力テンソルの空間次元に再配置します。
詳細な説明:
DepthToSpace 操作は、形状 [N, C, D1, D2, ..., DK]
を持つ入力テンソルの要素を出力テンソルに並べ替えます。出力テンソルでは、入力深度次元 (特徴) C
の値が D1
, …, DK
の空間ブロックに移動されます。
この操作は、形状 [N, C, D1, D2, ..., DK]
の K
空間次元を持つ入力テンソル data
を Y 出力テンソルに変換する次の変換と等価です。mode = blocks_first
の場合:
x' = reshape(data, [N, block_size, block_size, ..., block_size, C / (block_size ^ K), D1, D2, ..., DK])
x'' = transpose(x', [0, K + 1, K + 2, 1, K + 3, 2, K + 4, 3, ..., K + (K + 1), K])
y = reshape(x'', [N, C / (block_size ^ K), D1 * block_size, D2 * block_size, D3 * block_size, ..., DK * block_size])
mode = depth_first
の場合:
x' = reshape(data, [N, C / (block_size ^ K), block_size, block_size, ..., block_size, D1, D2, ..., DK])
x'' = transpose(x', [0, 1, K + 2, 2, K + 3, 3, K + 4, 4, ..., K + (K + 1), K + 1])
y = reshape(x'', [N, C / (block_size ^ K), D1 * block_size, D2 * block_size, D3 * block_size, ..., DK * block_size])
属性:
-
block_size
説明: 移動する値ブロックのサイズを指定します。深度の次元のサイズは、
block_size ^ (len(input.shape) - 2)
で均等に分割する必要があります。値の範囲: 正の整数
タイプ:
int
デフォルト値: 1
必須: いいえ
-
mode
説明: 入力深度の次元をブロック座標と新しい深度の次元に分割する方法を指定します。
-
値の範囲:
blocks_first: 入力深度は次のように分割されます
[block_size, ..., block_size, new_depth]
depth_first: 入力深度は次のように分割されます
[new_depth, 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="DepthToSpace" ...>
<data block_size="2" mode="blocks_first"/>
<input>
<port id="0">
<dim>5</dim>
<dim>28</dim>
<dim>2</dim>
<dim>3</dim>
</port>
</input>
<output>
<port id="1">
<dim>5</dim> <!-- data.shape[0] -->
<dim>7</dim> <!-- data.shape[1] / (block_size ^ 2) -->
<dim>4</dim> <!-- data.shape[2] * block_size -->
<dim>6</dim> <!-- data.shape[3] * block_size -->
</port>
</output>
</layer>