টেনসরফ্লো :: অপস:: সংগ্রহ ভি 2
#include <array_ops.h>
indices
অনুযায়ী params
অক্ষ axis
থেকে স্লাইস সংগ্রহ করুন ।
সারাংশ
indices
অবশ্যই যেকোনো মাত্রার একটি পূর্ণসংখ্যার টেনসর হতে হবে (সাধারণত 0-D বা 1-D)। আকৃতি params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]
সহ একটি আউটপুট টেনসর তৈরি করে যেখানে:
# 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]
মনে রাখবেন যে CPU-তে, আউট অফ বাউন্ড সূচক পাওয়া গেলে, একটি ত্রুটি ফেরত দেওয়া হয়। GPU-তে, যদি একটি আউট অফ বাউন্ড সূচক পাওয়া যায়, একটি 0 সংশ্লিষ্ট আউটপুট মানের মধ্যে সংরক্ষণ করা হয়।
এছাড়াও tf.batch_gather
এবং tf.gather_nd
দেখুন।
যুক্তি:
- স্কোপ: একটি স্কোপ অবজেক্ট
- params: যে টেনসর থেকে মান সংগ্রহ করতে হয়। কমপক্ষে র্যাঙ্ক
axis + 1
হতে হবে। - indices: সূচক টেনসর। পরিসরে হতে হবে
[0, params.shape[axis])
। - axis:
params
থাকা অক্ষ থেকেindices
সংগ্রহ করতে হবে। প্রথম মাত্রা ডিফল্ট. নেতিবাচক সূচক সমর্থন করে।
রিটার্ন:
-
Output
:indices
দ্বারা প্রদত্ত সূচকগুলি থেকে সংগ্রহ করাparams
থেকে মানগুলি, আকারparams.shape[:axis] + indices.shape + params.shape[axis + 1:]
সহ।
কনস্ট্রাক্টর এবং ডেস্ট্রাক্টর | |
---|---|
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) |
পাবলিক বৈশিষ্ট্য | |
---|---|
operation | |
output |
পাবলিক ফাংশন | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
পাবলিক স্ট্যাটিক ফাংশন | |
---|---|
BatchDims (int64 x) |
কাঠামো | |
---|---|
tensorflow:: ops:: GatherV2:: Attrs | GatherV2 এর জন্য ঐচ্ছিক অ্যাট্রিবিউট সেটার। |
পাবলিক বৈশিষ্ট্য
অপারেশন
Operation operation
আউটপুট
::tensorflow::Output output
পাবলিক ফাংশন
সংগ্রহ ভি 2
GatherV2( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, ::tensorflow::Input axis )
সংগ্রহ ভি 2
GatherV2( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, ::tensorflow::Input axis, const GatherV2::Attrs & attrs )
নোড
::tensorflow::Node * node() const
অপারেটর::টেনসরফ্লো::ইনপুট
operator::tensorflow::Input() const
অপারেটর::টেনসরফ্লো::আউটপুট
operator::tensorflow::Output() const
পাবলিক স্ট্যাটিক ফাংশন
ব্যাচডিমস
Attrs BatchDims( int64 x )