aliran tensor:: operasi:: KumpulkanV2
#include <array_ops.h>
Kumpulkan irisan dari axis
params
sesuai dengan indices
.
Ringkasan
indices
harus berupa tensor bilangan bulat dengan dimensi apa pun (biasanya 0-D atau 1-D). Menghasilkan tensor keluaran dengan bentuk params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]
di mana:
# Scalar indices (output is rank(params) - 1). output[a_0, ..., a_n, b_0, ..., b_n] = params[a_0, ..., a_n, indices, b_0, ..., b_n]
# Vector indices (output is rank(params)). output[a_0, ..., a_n, i, b_0, ..., b_n] = params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
# Higher rank indices (output is rank(params) + rank(indices) - 1). output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] = params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
Perhatikan bahwa pada CPU, jika indeks di luar batas ditemukan, kesalahan akan dikembalikan. Pada GPU, jika indeks di luar batas ditemukan, angka 0 disimpan dalam nilai keluaran yang sesuai.
Lihat juga tf.batch_gather
dan tf.gather_nd
.
Argumen:
- ruang lingkup: Objek Lingkup
- params: Tensor untuk mengumpulkan nilai. Setidaknya harus memiliki peringkat
axis + 1
. - indeks: Tensor indeks. Harus berada dalam rentang
[0, params.shape[axis])
. - axis: Sumbu dalam
params
untuk mengumpulkanindices
. Default ke dimensi pertama. Mendukung indeks negatif.
Pengembalian:
-
Output
: Nilai dariparams
yang dikumpulkan dari indeks yang diberikan olehindices
, dengan bentukparams.shape[:axis] + indices.shape + params.shape[axis + 1:]
.
Konstruktor dan Destruktor | |
---|---|
GatherV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, :: tensorflow::Input axis) | |
GatherV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, :: tensorflow::Input axis, const GatherV2::Attrs & attrs) |
Atribut publik | |
---|---|
operation | |
output |
Fungsi publik | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Fungsi statis publik | |
---|---|
BatchDims (int64 x) |
Struktur | |
---|---|
tensorflow:: ops:: GatherV2:: Attrs | Penyetel atribut opsional untuk GatherV2 . |
Atribut publik
operasi
Operation operation
keluaran
::tensorflow::Output output
Fungsi publik
KumpulkanV2
GatherV2( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, ::tensorflow::Input axis )
KumpulkanV2
GatherV2( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, ::tensorflow::Input axis, const GatherV2::Attrs & attrs )
simpul
::tensorflow::Node * node() const
operator::tensorflow::Masukan
operator::tensorflow::Input() const
operator::tensorflow::Keluaran
operator::tensorflow::Output() const
Fungsi statis publik
BatchDim
Attrs BatchDims( int64 x )