このページは Cloud Translation API によって翻訳されました。
Switch to English

tf.stack

TensorFlow 1つのバージョン GitHubの上のソースを表示

rank-のリストスタックR 1 rank-にテンソルを(R+1)テンソル。

ノートPCで使用されます

ガイドで使用チュートリアルで使用されます

参照してくださいtf.concattf.tiletf.repeat

パックでのテンソルのリストvalues各テンソルよりも高いランク1とテンソルへのvaluesに沿ってそれらを充填することにより、 axisのディメンション。長さのリストを与えられたN形のテンソルの(A, B, C)

場合axis == 0 、次にoutputテンソルの形状があります(N, A, B, C) 。場合axis == 1次にoutputテンソルの形状があります(A, N, B, C) 。等。

例えば:

x = tf.constant([1, 4])
y = tf.constant([2, 5])
z = tf.constant([3, 6])
tf.stack([x, y, z])
<tf.Tensor: shape=(3, 2), dtype=int32, numpy=
array([[1, 4],
       [2, 5],
       [3, 6]], dtype=int32)>
tf.stack([x, y, z], axis=1)
<tf.Tensor: shape=(2, 3), dtype=int32, numpy=
array([[1, 2, 3],
       [4, 5, 6]], dtype=int32)>

これは、スタック解除の反対です。 numpyのと等価であるnp.stack

np.array_equal(np.stack([x, y, z]), tf.stack([x, y, z]))
True

values リストTensor同じ形状およびタイプのオブジェクト。
axis int 。軸に沿ってスタックに。最初の次元にデフォルト設定。有効範囲があるので、負の値は、ラップアラウンド[-(R+1), R+1)
name この操作の名前(オプション)。

output Aは、積み重ねられたTensor同じタイプのvalues

ValueError 場合axis範囲外である[ - (R + 1)、R + 1)。