ML Topluluk Günü 9 Kasım! TensorFlow, JAX güncellemeler için bize katılın ve daha fazla bilgi edinin

tensorflow :: ops :: Topla

#include <array_ops.h>

indices göre params dilimler toplayın .

Özet

indices , herhangi bir boyutun (genellikle 0-D veya 1-D) bir tamsayı tensörü olmalıdır. Şekil indices.shape + params.shape[1:] şeklinde bir çıktı tensörü üretir, burada:

    # Scalar indices
    output[:, ..., :] = params[indices, :, ... :]

    # Vector indices
    output[i, :, ..., :] = params[indices[i], :, ... :]

    # Higher rank indices
    output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]

indices bir permütasyon ise ve len(indices) == params.shape[0] bu işlem uygun şekilde params verecektir.

validate_indices : KULLANIMDAN KALDIRILDI. Bu işlem CPU'ya atanırsa, indices değerlerin aralık dahilinde olduğu her zaman doğrulanır. GPU'ya atanırsa, sınır dışı endeksler güvenli ancak belirsiz davranışla sonuçlanır; bu, bir hata oluşturmayı da içerebilir.

Argümanlar:

  • kapsam: Bir Scope nesnesi

İadeler:

  • Output : Çıkış tensörü.

Yapıcılar ve Yıkıcılar

Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices)
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, const Gather::Attrs & attrs)

Genel özellikler

operation
output

Kamusal işlevler

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Genel statik işlevler

ValidateIndices (bool x)

Yapılar

tensorflow :: ops :: Gather :: Attrs

Gather için isteğe bağlı öznitelik belirleyiciler.

Genel özellikler

operasyon

Operation operation

çıktı

::tensorflow::Output output

Kamusal işlevler

Topla

 Gather(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices
)

Topla

 Gather(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  const Gather::Attrs & attrs
)

düğüm

::tensorflow::Node * node() const 

operator :: tensorflow :: Girdi

 operator::tensorflow::Input() const 

operator :: tensorflow :: Çıktı

 operator::tensorflow::Output() const 

Genel statik işlevler

Endeksleri Doğrula

Attrs ValidateIndices(
  bool x
)