Juntar

público final de aula Gather

Reúna fatias do eixo `params` e` eixo` de acordo com os `índices`.

`índices` deve ser um tensor inteiro de qualquer dimensão (geralmente 0-D ou 1-D). Produz um tensor de saída com forma `params.shape [: axis] + indices.shape [batch_dims:] + params.shape [axis + 1:]` onde:

# 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]
 

Observe que, na CPU, se um índice fora do limite for encontrado, um erro será retornado. Na GPU, se um índice fora do limite for encontrado, um 0 é armazenado no valor de saída correspondente.

Veja também `tf.batch_gather` e` tf.gather_nd`.

Classes aninhadas

aula

Gather.Options

Atributos opcionais paraGather

Métodos Públicos

Saída <T>	asOutput () Retorna o identificador simbólico de um tensor.
static Gather.Options	batchDims (long batchDims)
static <T, U extends Number, V extends Number> Gather <T>	criar ( Scope escopo, Operando <T> params, Operando <U> índices, Operando <V> eixo, Options ... Opções) Método de fábrica para criar uma classe que envolve uma nova operação Gather.
Saída <T>	saída () Valores de `params` coletados de índices dados por` índices`, com forma `params.shape [: axis] + indices.shape + params.shape [axis + 1:]`.

Métodos herdados

Da classe org.tensorflow.op.PrimitiveOp

final booleano	igual a (objeto obj)
final int	hashCode ()
Operação	op () Retorna a Operation subjacente
corda final	toString ()

Da classe java.lang.Object

boleano	igual a (objeto arg0)
aula final <?>	getClass ()
int	hashCode ()
vazio final	notificar ()
vazio final	notificar tudo ()
Fragmento	toString ()
vazio final	espera (long arg0, int arg1)
vazio final	espera (long arg0)
vazio final	espera ()

Da interface org.tensorflow.Operand

Saída abstrata <T>

asOutput () Retorna o identificador simbólico de um tensor.

Métodos Públicos

public Output <T> asOutput ()

Retorna o identificador simbólico de um tensor.

As entradas para as operações do TensorFlow são saídas de outra operação do TensorFlow. Este método é usado para obter um identificador simbólico que representa o cálculo da entrada.

public static Gather.Options batchDims (Long batchDims)

public static Reúna <T> create ( Scope escopo, Operando <T> params, Operando <U> índices, Operando <V> eixo, Options ... Opções)

Método de fábrica para criar uma classe que envolve uma nova operação Gather.

Parâmetros

alcance	escopo atual
params	O tensor a partir do qual reunir valores. Deve ser pelo menos classificado como `eixo + 1`.
índices	Tensor de índice. Deve estar no intervalo `[0, params.shape [axis])`.
eixo	O eixo em `params` para reunir` índices`. O padrão é a primeira dimensão. Suporta índices negativos.
opções	carrega valores de atributos opcionais

Devoluções

• uma nova instância de Gather

output public Output <T> output ()

Valores de `params` coletados de índices dados por` indices`, com forma `params.shape [: axis] + indices.shape + params.shape [axis + 1:]`.