aliran tensor:: operasi:: Mengumpulkan
#include <array_ops.h>
Kumpulkan irisan dari params
menurut indices
.
Ringkasan
indices
harus berupa tensor bilangan bulat dari dimensi apa pun (biasanya 0-D atau 1-D). Menghasilkan tensor keluaran dengan bentuk indices.shape + params.shape[1:]
di mana:
# Scalar indices output[:, ..., :] = params[indices, :, ... :]
# Vector indices output[i, :, ..., :] = params[indices[i], :, ... :]
# Higher rank indices output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]
Jika indices
adalah permutasi dan len(indices) == params.shape[0]
maka operasi ini akan mengubah params
sesuai dengan itu.
validate_indices
: DIHAPUS. Jika operasi ini ditetapkan ke CPU, nilai dalam indices
selalu divalidasi agar berada dalam jangkauan. Jika ditetapkan ke GPU, indeks di luar batas menghasilkan perilaku yang aman tetapi tidak ditentukan, yang mungkin termasuk memunculkan kesalahan.
Argumen:
- ruang lingkup: Objek Lingkup
Pengembalian:
-
Output
: Tensor keluaran.
Konstruktor dan Destruktor | |
---|---|
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices) | |
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, const Gather::Attrs & attrs) |
Atribut publik | |
---|---|
operation | |
output |
Fungsi publik | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Fungsi statis publik | |
---|---|
ValidateIndices (bool x) |
Struktur | |
---|---|
tensorflow:: ops:: Kumpulkan:: Attrs | Setter atribut opsional untuk Gather . |
Atribut publik
operasi
Operation operation
keluaran
::tensorflow::Output output
Fungsi publik
Mengumpulkan
Gather( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices )
Mengumpulkan
Gather( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, const Gather::Attrs & attrs )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Fungsi statis publik
ValidasiIndeks
Attrs ValidateIndices( bool x )