Reúna cortes del eje `params` `eje` de acuerdo con `índices`.
`índices` debe ser un tensor entero de cualquier dimensión (generalmente 0-D o 1-D). Produce un tensor de salida con forma `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` donde:
# Scalar indices (output is rank(params) - 1).
output[a_0, ..., a_n, b_0, ..., b_n] =
params[a_0, ..., a_n, indices, b_0, ..., b_n]
# Vector indices (output is rank(params)).
output[a_0, ..., a_n, i, b_0, ..., b_n] =
params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
# Higher rank indices (output is rank(params) + rank(indices) - 1).
output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
Tenga en cuenta que en la CPU, si se encuentra un índice fuera de límite, se devuelve un error. En GPU, si se encuentra un índice fuera de límites, se almacena un 0 en el valor de salida correspondiente.
Consulte también `tf.batch_gather` y tf.gather_nd
.
Clases anidadas
clase | Reunir.Opciones | Atributos opcionales para Gather |
Métodos públicos
Salida <T> | como salida () Devuelve el identificador simbólico de un tensor. |
Gather.Options estáticas | batchDims (long batchDims) |
estático <T, U extiende Número, V extiende Número> Reunir <T> | |
Salida <T> | salida () Valores de `params` recopilados a partir de índices proporcionados por `indices`, con la forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`. |
Métodos Heredados
Métodos públicos
salida pública <T> como salida ()
Devuelve el identificador simbólico de un tensor.
Las entradas de las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
public static Gather <T> create ( alcance del alcance, parámetros del operando <T>, índices del operando <U>, eje del operando <V>, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación Gather.
Parámetros
alcance | alcance actual |
---|---|
parámetros | El tensor a partir del cual recopilar valores. Debe tener al menos el rango `eje + 1`. |
índices | Tensor índice. Debe estar en el rango `[0, params.shape[axis])`. |
eje | El eje en `params` para recopilar `índices`. El valor predeterminado es la primera dimensión. Admite índices negativos. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de Gather
salida pública <T> salida ()
Valores de `params` recopilados a partir de índices proporcionados por `indices`, con la forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.
Reúna cortes del eje `params` `eje` de acuerdo con `índices`.
`índices` debe ser un tensor entero de cualquier dimensión (generalmente 0-D o 1-D). Produce un tensor de salida con forma `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` donde:
# Scalar indices (output is rank(params) - 1).
output[a_0, ..., a_n, b_0, ..., b_n] =
params[a_0, ..., a_n, indices, b_0, ..., b_n]
# Vector indices (output is rank(params)).
output[a_0, ..., a_n, i, b_0, ..., b_n] =
params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
# Higher rank indices (output is rank(params) + rank(indices) - 1).
output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
Tenga en cuenta que en la CPU, si se encuentra un índice fuera de límite, se devuelve un error. En GPU, si se encuentra un índice fuera de límites, se almacena un 0 en el valor de salida correspondiente.
Consulte también `tf.batch_gather` y tf.gather_nd
.
Clases anidadas
clase | Reunir.Opciones | Atributos opcionales para Gather |
Métodos públicos
Salida <T> | como salida () Devuelve el identificador simbólico de un tensor. |
Gather.Options estáticas | batchDims (long batchDims) |
estático <T, U extiende Número, V extiende Número> Reunir <T> | |
Salida <T> | salida () Valores de `params` recopilados a partir de índices proporcionados por `indices`, con la forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`. |
Métodos Heredados
Métodos públicos
salida pública <T> como salida ()
Devuelve el identificador simbólico de un tensor.
Las entradas de las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.
public static Gather <T> create ( alcance del alcance, parámetros del operando <T>, índices del operando <U>, eje del operando <V>, opciones... opciones)
Método de fábrica para crear una clase que envuelve una nueva operación Gather.
Parámetros
alcance | alcance actual |
---|---|
parámetros | El tensor a partir del cual recopilar valores. Debe tener al menos el rango `eje + 1`. |
índices | Tensor índice. Debe estar en el rango `[0, params.shape[axis])`. |
eje | El eje en `params` para recopilar `índices`. El valor predeterminado es la primera dimensión. Admite índices negativos. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de Gather
salida pública <T> salida ()
Valores de `params` recopilados a partir de índices proporcionados por `indices`, con la forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.