Esta página foi traduzida pela API Cloud Translation.
Switch to English

tf.IndexedSlices

TensorFlow 1 versão Ver fonte no GitHub

Uma representação esparsa de um conjunto de fatias de tensores em índices dadas.

Esta classe é um wrapper simples para um par de Tensor objetos:

  • values : Um Tensor de qualquer forma com dtipo [D0, D1, ..., Dn] .
  • indices : A 1-D inteiro Tensor com forma [D0] .

Um IndexedSlices é tipicamente utilizado para representar um subconjunto de um tensor de maior dense de forma [LARGE0, D1, .. , DN] onde LARGE0 >> D0 . Os valores em indices são os índices na primeira dimensão das fatias que foram extraídos a partir do tensor maior.

O tensor densa dense representada por um IndexedSlices slices tem

 dense[slices.indices[i], :, :, :, ...] = slices.values[i, :, :, :, ...]
 

O IndexedSlices classe é usada principalmente na definição de gradientes para operações que tenham gradientes esparsos (por exemplo tf.gather ).

Contraste esta representação com tf.sparse.SparseTensor , que utiliza índices multi-dimensionais e valores escalares.

dense_shape Um 1-D Tensor contendo a forma do tensor densa correspondente.
device O nome do dispositivo no qual values serão produzidos, ou None .
dtype O DType de elementos nesta tensor.
graph O Graph que contém os valores, índices e tensores de forma.
indices Um 1-D Tensor contendo os índices das fatias.
name O nome desta IndexedSlices .
op A Operation que produz values como uma saída.
shape Obtém o tf.TensorShape representando a forma do tensor densa.
values Um Tensor contendo os valores das fatias.

Métodos

consumers

Ver fonte

__neg__

Ver fonte