aliran tensor:: operasi:: Perpecahan Jarang

Pisahkan SparseTensor menjadi num_split tensor sepanjang satu dimensi.

Ringkasan

Jika shape[split_dim] bukan kelipatan bilangan bulat dari num_split . Irisan [0 : shape[split_dim] % num_split] mendapat satu dimensi tambahan. Misalnya jika split_dim = 1 dan num_split = 2 dan inputnya adalah

input_tensor = shape = [2, 7]
[    a   d e  ]
[b c          ]

Secara grafis tensor keluarannya adalah:

output_tensor[0] = shape = [2, 4]
[    a  ]
[b c    ]

output_tensor[1] = shape = [2, 3]
[ d e  ]
[      ]

Argumen:

• ruang lingkup: Objek Lingkup
• split_dim: 0-D. Dimensi yang akan dipecah. Harus berada dalam rentang [0, rank(shape)) .
• indeks: tensor 2-D mewakili indeks tensor renggang.
• nilai: Tensor 1-D mewakili nilai tensor renggang.
• bentuk: 1-D. tensor mewakili bentuk tensor renggang. indeks keluaran: Daftar tensor 1-D mewakili indeks tensor renggang keluaran.
• num_split: Banyaknya cara untuk membagi.

Pengembalian:

• OutputList keluaran_indeks
• OutputList output_values: Daftar tensor 1-D mewakili nilai tensor renggang keluaran.
• OutputList output_shape: Daftar tensor 1-D mewakili bentuk tensor renggang keluaran.

Atribut publik

Operation operation

::tensorflow::OutputList output_indices

::tensorflow::OutputList output_shape

::tensorflow::OutputList output_values

 SparseSplit(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input split_dim,
  ::tensorflow::Input indices,
  ::tensorflow::Input values,
  ::tensorflow::Input shape,
  int64 num_split
)