টেনসরফ্লো :: অপস:: ডাইনামিক পার্টিশন
#include <data_flow_ops.h>
partitions
থেকে সূচক ব্যবহার করে num_partitions
টেনসরে data
পার্টিশন করে।
সারাংশ
partitions.ndim
আকারের প্রতিটি ইনডেক্স টিপল js
এর জন্য, স্লাইস data[js, ...]
outputs[partitions[js]]
এর অংশ হয়ে যায়। partitions[js] = i
outputs[i]
js
এর অভিধানিক ক্রমানুসারে স্থাপন করা হয় এবং outputs[i]
হল i
এর সমান partitions
এন্ট্রির সংখ্যা। বিস্তারিতভাবে,
outputs[i].shape = [sum(partitions == i)] + data.shape[partitions.ndim:]
outputs[i] = pack([data[js, ...] for js if partitions[js] == i])
data.shape
অবশ্যই partitions.shape
দিয়ে শুরু করতে হবে।
যেমন:
# Scalar partitions. partitions = 1 num_partitions = 2 data = [10, 20] outputs[0] = [] # Empty with shape [0, 2] outputs[1] = [[10, 20]]
# Vector partitions. partitions = [0, 0, 1, 1, 0] num_partitions = 2 data = [10, 20, 30, 40, 50] outputs[0] = [10, 20, 50] outputs[1] = [30, 40]
কিভাবে পার্টিশন আবার মার্জ করতে হয় তার উদাহরণের জন্য dynamic_stitch
দেখুন।
যুক্তি:
- স্কোপ: একটি স্কোপ অবজেক্ট
- পার্টিশন: যেকোনো আকৃতি। পরিসরে সূচকগুলি
[0, num_partitions)
। - num_partitions: আউটপুটে পার্টিশনের সংখ্যা।
রিটার্ন:
-
OutputList
: আউটপুট টেনসর।
কনস্ট্রাক্টর এবং ডেস্ট্রাক্টর | |
---|---|
DynamicPartition (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input partitions, int64 num_partitions) |
পাবলিক বৈশিষ্ট্য | |
---|---|
operation | |
outputs |
পাবলিক ফাংশন | |
---|---|
operator[] (size_t index) const |
পাবলিক বৈশিষ্ট্য
অপারেশন
Operation operation
আউটপুট
::tensorflow::OutputList outputs
পাবলিক ফাংশন
ডাইনামিক পার্টিশন
DynamicPartition( const ::tensorflow::Scope & scope, ::tensorflow::Input data, ::tensorflow::Input partitions, int64 num_partitions )
অপারেটর
::tensorflow::Output operator[]( size_t index ) const