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

tensorflow :: ops :: Seyrek Ekle

#include <sparse_ops.h>

Başka bir SparseTensor üretmek için iki SparseTensor nesnesi ekler.

Özet

Giriş SparseTensor nesnelerinin dizinlerinin standart sözlük sırasına göre sıralandığı varsayılır. Durum böyle değilse, bu adımdan önce dizin SparseReorder geri yüklemek için SparseReorder çalıştırın.

Varsayılan olarak, bir dizinde iki değerin toplamı sıfırsa, çıktı SparseTensor yine de dizininde bu belirli konumu içerecek ve karşılık gelen değer yuvasında bir sıfırı depolayacaktır. Bunu geçersiz kılmak için, arayanlar, toplamın thresh kesinlikle daha küçük bir büyüklüğe sahip olması durumunda, karşılık gelen değerinin ve endeksinin dahil edilmeyeceğini belirten bir thresh belirleyebilir. Özellikle, eşik thresh == 0 (varsayılan), her şeyin korunduğu ve gerçek eşiklemenin yalnızca pozitif bir değer için gerçekleştiği anlamına gelir.

Aşağıdaki şekiller olarak, nnz aldıktan sonra sayısıdır thresh dikkate.

Argümanlar:

  • kapsam: Bir Scope nesnesi
  • a_indices: 2-D. indices ilk SparseTensor , boyut [nnz, ndims] matris.
  • a_values: 1-D. İlk SparseTensor values , boyut [nnz] Vektörü.
  • a_shape: 1-D. İlk SparseTensor shape , boyut [ndims] Vector.
  • b_indices: 2-D. indices ikinci SparseTensor , boyut [nnz, ndims] matris.
  • b_values: 1-D. İkinci SparseTensor values , boyut [nnz] Vektörü.
  • b_shape: 1-D. İkinci SparseTensor shape , boyut [ndims] Vektörü.
  • eşik: 0-D. Bir çıktı değeri / dizin çiftinin yer kaplayıp kaplamadığını belirleyen büyüklük eşiği.

İadeler:

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

SparseAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b_indices, :: tensorflow::Input b_values, :: tensorflow::Input b_shape, :: tensorflow::Input thresh)

Genel özellikler

operation
sum_indices
sum_shape
sum_values

Genel özellikler

operasyon

Operation operation

sum_indices

::tensorflow::Output sum_indices

sum_shape

::tensorflow::Output sum_shape

toplam_değerler

::tensorflow::Output sum_values

Kamusal işlevler

Seyrek Ekle

 SparseAdd(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input a_values,
  ::tensorflow::Input a_shape,
  ::tensorflow::Input b_indices,
  ::tensorflow::Input b_values,
  ::tensorflow::Input b_shape,
  ::tensorflow::Input thresh
)