Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

XlaSplitND

public class akhir XlaSplitND

Membagi tensor input di semua dimensi.

Sebuah op yang mengiris tensor input berdasarkan atribut num_splits yang diberikan, pads mengiris secara opsional, dan mengembalikan irisan. 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]]
 
` num_splits`:
[2, 2]
 
dan `paddings`:
[1, 1]
 
yang diharapkan` outputs` adalah:
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]
 

Kelas Bersarang

kelas XlaSplitND.Options Atribut opsional untuk XlaSplitND

Metode Publik

statis <T> XlaSplitND <T>
membuat ( Lingkup lingkup, Operan <T> input, panjang N, Daftar <Panjang> numSplits, Options ... pilihan)
Metode pabrik untuk membuat kelas yang membungkus operasi XlaSplitND baru.
Iterator < Operan <T >>
Daftar < Keluaran <T >>
output ()
statis XlaSplitND.Options
paddings (Daftar <Panjang> paddings)

Metode yang Diwarisi

Metode Publik

public static XlaSplitND <T> membuat ( Lingkup lingkup, Operan <T> input, panjang N, Daftar <Panjang> numSplits, Options ... pilihan)

Metode pabrik untuk membuat kelas yang membungkus operasi XlaSplitND baru.

Parameter
cakupan lingkup saat ini
memasukkan Masukkan tensor untuk membagi di semua dimensi. } out_arg { nama: "keluaran" deskripsi: <
numSplits Jumlah cara untuk membagi per dimensi. Dimensi bentuk harus dapat dibagi secara merata.
pilihan membawa nilai atribut opsional
Kembali
  • contoh baru XlaSplitND

public Iterator < Operan <T >> iterator ()

public List < Keluaran <T >> output ()

public static XlaSplitND.Options paddings (Daftar <Panjang> paddings)

Parameter
bantalan Daftar opsional padding kanan per dimensi tensor input untuk diterapkan sebelum pemisahan. Ini dapat digunakan untuk membuat dimensi dapat dibagi secara merata.