XlaSplitND
Tetap teratur dengan koleksi
Simpan dan kategorikan konten berdasarkan preferensi Anda.
Membagi tensor masukan di semua dimensi.
Sebuah operasi yang mengiris tensor masukan berdasarkan atribut num_splits yang diberikan, memasukkan irisan secara opsional, dan mengembalikan irisan tersebut. Irisan dikembalikan dalam urutan baris-utama.
Operasi ini dapat dihasilkan melalui jembatan TPU.
Misalnya, dengan tensor `input`:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
`jumlah_pemisahan`:
[2, 2]
dan `padding`:
[1, 1]
`output` yang diharapkan adalah:
[[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Metode Warisan
Dari kelas java.lang.Object boolean | sama dengan (Objek arg0) |
Kelas terakhir<?> | dapatkan Kelas () |
ke dalam | Kode hash () |
kekosongan terakhir | beri tahu () |
kekosongan terakhir | beri tahuSemua () |
Rangkaian | keString () |
kekosongan terakhir | tunggu (arg0 panjang, int arg1) |
kekosongan terakhir | tunggu (argumen panjang0) |
kekosongan terakhir | Tunggu () |
Dari antarmuka java.lang.Iterable ruang kosong | forEach (Konsumen<? super T> arg0) |
abstrak Iterator< Operan <T>> | pengulangan () |
Pemisah< Operan <T>> | pemisah () |
Metode Publik
Metode pabrik untuk membuat kelas yang membungkus operasi XlaSplitND baru.
Parameter
cakupan | ruang lingkup saat ini |
---|
masukan | Masukkan tensor untuk membagi seluruh dimensi. } out_arg { nama: "keluaran" deskripsi: < |
---|
nomorSplit | Banyaknya cara untuk membagi per dimensi. Dimensi bentuk harus dapat dibagi rata. |
---|
pilihan | membawa nilai atribut opsional |
---|
Iterator publik< Operan <T>> iterator ()
Daftar publik< Keluaran <T>> keluaran ()
padding XlaSplitND.Options statis publik (padding Daftar<Panjang>)
Parameter
bantalan | Daftar opsional bantalan kanan per dimensi tensor masukan yang akan diterapkan sebelum pemisahan. Ini dapat digunakan untuk membuat suatu dimensi dapat dibagi rata. |
---|
Kecuali dinyatakan lain, konten di halaman ini dilisensikan berdasarkan Lisensi Creative Commons Attribution 4.0, sedangkan contoh kode dilisensikan berdasarkan Lisensi Apache 2.0. Untuk mengetahui informasi selengkapnya, baca Kebijakan Situs Google Developers. Java adalah merek dagang terdaftar dari Oracle dan/atau afiliasinya. Beberapa konten dilisensikan berdasarkan lisensi numpy.
Terakhir diperbarui pada 2024-08-24 UTC.
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Informasi yang saya butuhkan tidak ada"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Terlalu rumit/langkahnya terlalu banyak"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Sudah usang"
},{
"type": "thumb-down",
"id": "translationIssue",
"label":"Masalah terjemahan"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Masalah kode / contoh"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Lainnya"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Mudah dipahami"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Memecahkan masalah saya"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Lainnya"
}]
{
"lastModified": "Terakhir diperbarui pada 2024-08-24 UTC.",
"confidential": False
}