SegmentProdV2

SegmentProdV2 kelas akhir publik

Menghitung produk sepanjang segmen tensor.

Baca [bagian segmentasi](https://tensorflow.org/api_docs/python/tf/math#Segmentation) untuk penjelasan tentang segmen.

Menghitung tensor sedemikian rupa sehingga \\(output_i = \prod_j data_j\\) dimana produknya melebihi `j` sehingga `segment_ids[j] == i`.

Jika produk kosong untuk ID segmen tertentu `i`, `output[i] = 1`.

Catatan: Operasi ini saat ini hanya didukung dengan jit_compile=True.

Satu-satunya perbedaan dengan SegmentProd adalah input tambahan `num_segments`. Ini membantu dalam mengevaluasi bentuk keluaran dalam waktu kompilasi. `num_segments` harus konsisten dengan segment_ids. misalnya Maks(segment_ids) - 1 harus sama dengan `num_segments` untuk segment_ids 1 hari Dengan num_segments yang tidak konsisten, operasi tetap berjalan. satu-satunya perbedaan adalah, output mengambil ukuran num_segments terlepas dari ukuran segment_ids dan data. untuk num_segments lebih kecil dari ukuran keluaran yang diharapkan, elemen terakhir diabaikan, untuk num_segments lebih besar dari ukuran keluaran yang diharapkan, elemen terakhir diberi 1.

Misalnya:

>>> @tf.function(jit_compile=True) ... def test(c): ... mengembalikan tf.raw_ops.SegmentProdV2(data=c, segment_ids=tf.constant([0, 0, 1]), jumlah_segmen=2) >>> c = tf.constant([[1,2,3,4], [4, 3, 2, 1], [5,6,7,8]]) >>> tes( c).numpy() larik([[4, 6, 6, 4], [5, 6, 7, 8]], tiped=int32)

Metode Publik

Keluaran <T>
sebagai Keluaran ()
Mengembalikan pegangan simbolik tensor.
statis <T, U memperluas Nomor, V memperluas Nomor> SegmentProdV2 <T>
buat ( Lingkup lingkup , Operand <T> data, Operand <U> segmentIds, Operand <V> numSegments)
Metode pabrik untuk membuat kelas yang membungkus operasi SegmentProdV2 baru.
Keluaran <T>
keluaran ()
Bentuknya sama dengan data, kecuali dimensi `segment_ids.rank` pertama, yang diganti dengan dimensi tunggal yang berukuran `num_segments`.

Metode Warisan

Metode Publik

Keluaran publik <T> sebagai Keluaran ()

Mengembalikan pegangan simbolik tensor.

Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.

public static SegmentProdV2 <T> buat ( Lingkup cakupan , Operand <T> data, Operand <U> segmentIds, Operand <V> numSegments)

Metode pabrik untuk membuat kelas yang membungkus operasi SegmentProdV2 baru.

Parameter
cakupan ruang lingkup saat ini
segmenId Tensor 1-D yang ukurannya sama dengan ukuran dimensi pertama `data`. Nilai harus diurutkan dan dapat diulang. Nilainya harus kurang dari `num_segments`.

Perhatian: Nilai selalu divalidasi untuk diurutkan pada CPU, tidak pernah divalidasi pada GPU.

Kembali
  • contoh baru dari SegmentProdV2

Keluaran publik <T> keluaran ()

Bentuknya sama dengan data, kecuali dimensi `segment_ids.rank` pertama, yang diganti dengan dimensi tunggal yang berukuran `num_segments`.