TensorFlow函數(shù):IndexedSlices

2018-10-26 17:33 更新
函數(shù):tf.IndexedSlices

IndexedSlices 類

定義在:tensorflow/python/framework/ops.py.

參見(jiàn)指南:變量>稀疏變量更新

在給定索引上的一組張量切片的稀疏表示.
此類是一對(duì)張量對(duì)象的簡(jiǎn)單包裝:

  • values:具有形狀 [D0, D1, ..., Dn] 的任何類型的張量.
  • indices:具有形狀 [D0] 的一維整數(shù)張量.

IndexedSlices 通常用于表示較大張量形狀為 [LARGE0, D1,.., DN] 的子集,其中 LARGE0 >> D0.指數(shù)中的值是從較大張量中提取的切片的第一個(gè)維度中的索引.

由 IndexedSlices 切片表示的稠密張量有:

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

IndexedSlices 類主要用于具有稀疏漸變 (如 tf.gather) 的操作的漸變定義.

將此表示與 tf.SparseTensor 進(jìn)行對(duì)比,它使用多維指數(shù)和標(biāo)量值.

屬性

  • dense_shape
    包含相應(yīng)稠密張量的形狀的1維張量.
  • device
    將在其上生成值的設(shè)備的名稱,或 None.
  • dtype
    此張量中元素的 DType.
  • graph
    包含值、索引和形狀張量的圖.
  • indices
    包含切片的索引的1維張量.
  • name
    該 IndexedSlices 的名字.
  • op
    將值作為輸出生成的操作.
  • values
    包含切片值的張量.

方法

__init__

__init__(
    values,
    indices,
    dense_shape=None
)

創(chuàng)建一個(gè) IndexedSlices.

__neg__

__neg__()
以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)