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\\) yang produknya berada di atas `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> | |
| 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`.