Se usó la API de Cloud Translation para traducir esta página.
Switch to English

tf.stack

TensorFlow 1 versión Ver código fuente en GitHub

Se acumula una lista de Rank- R tensores en una Rank- (R+1) tensor.

Se utiliza en los cuadernos

Se utiliza en la guía Se utiliza en los tutoriales

Ver también tf.concat , tf.tile , tf.repeat .

Packs La lista de los tensores en values en un tensor con rango uno más alto que cada tensor en values , por el embalaje a lo largo del axis dimensión. Dada una lista de longitud N de los tensores de forma (A, B, C) ;

Si axis == 0 entonces la output tensor tendrá la forma (N, A, B, C) . Si axis == 1 entonces la output tensor tendrá la forma (A, N, B, C) . Etcétera

Por ejemplo:

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)>

Esto es lo contrario de desapilamiento. El numpy equivalente es np.stack

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

values Una lista de Tensor objetos con la misma forma y tipo.
axis Un int . El eje de la pila a lo largo. El valor predeterminado es la primera dimensión. Los valores negativos se envuelven alrededor, por lo que el rango válido es [-(R+1), R+1) .
name Un nombre para esta operación (opcional).

output Un apilar Tensor con el mismo tipo como values .

ValueError Si axis está fuera de la gama [- (R + 1), R + 1).