কোডের পোর্টিং সহজ করে যা tf.nn.embedding_lookup() ব্যবহার করে।
নমুনা_বিভাজন[i], এম্বেডিং_সূচক[i] এবং সমষ্টি_ওজন[i] ith বৈশিষ্ট্যের সাথে মিলে যায়। table_ids[i] নির্দেশ করে কোন এম্বেডিং টেবিলটি ith বৈশিষ্ট্য দেখতে হবে।
দুটি ইনপুট তালিকা, এমবেডিং_সূচক এবং একত্রীকরণ_ওজনে সংশ্লিষ্ট অবস্থানে থাকা টেনসরগুলির অবশ্যই একই আকৃতি থাকতে হবে, যেমন অনুরূপ বৈশিষ্ট্য দ্বারা বর্ণিত টেবিলে মোট লুকআপের সংখ্যার সমান dim_size() সহ র্যাঙ্ক 1।
নেস্টেড ক্লাস
ক্লাস | EnqueueTPUEmbeddingRaggedTensorBatch.Options | জন্য ঐচ্ছিক বৈশিষ্ট্য EnqueueTPUEmbeddingRaggedTensorBatch |
পাবলিক পদ্ধতি
স্ট্যাটিক EnqueueTPUEmbeddingRaggedTensorBatch.Options | combiners (তালিকা <স্ট্রিং> combiners) |
স্ট্যাটিক <টি প্রসারিত নম্বর, ইউ নম্বর প্রসারিত, ভি প্রসারিত number> EnqueueTPUEmbeddingRaggedTensorBatch | তৈরি ( ব্যাপ্তি সুযোগ Iterable < প্রতীক <টি >> sampleSplits, Iterable < প্রতীক <u >> embeddingIndices, Iterable < প্রতীক <ভী >> aggregationWeights, প্রতীক <স্ট্রিং> modeOverride, তালিকা <লং> tableIds, বিকল্পসমূহ ... অপশন) একটি নতুন EnqueueTPUEmbeddingRaggedTensorBatch অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানার পদ্ধতি। |
স্ট্যাটিক EnqueueTPUEmbeddingRaggedTensorBatch.Options | deviceOrdinal (দীর্ঘ deviceOrdinal) |
স্ট্যাটিক EnqueueTPUEmbeddingRaggedTensorBatch.Options | maxSequenceLengths (তালিকা <লং> maxSequenceLengths) |
স্ট্যাটিক EnqueueTPUEmbeddingRaggedTensorBatch.Options | numFeatures (তালিকা <লং> numFeatures) |
উত্তরাধিকারসূত্রে প্রাপ্ত পদ্ধতি
পাবলিক পদ্ধতি
পাবলিক স্ট্যাটিক EnqueueTPUEmbeddingRaggedTensorBatch.Options combiners (তালিকা <স্ট্রিং> combiners)
পরামিতি
সংযোজক | স্ট্রিং স্কেলারগুলির একটি তালিকা, প্রতিটি এম্বেডিং টেবিলের জন্য একটি যা নির্দিষ্ট করে কিভাবে ওজনযুক্ত যোগফলের পরে এমবেডিং অ্যাক্টিভেশনগুলিকে স্বাভাবিক করা যায়। সমর্থিত কম্বাইনারগুলি হল 'মান', 'সমর্থন', বা 'sqrtn'। 'মান'-এর জন্য ওজনের যোগফল 0 বা 'sqrtn'-এর জন্য বর্গ ওজনের যোগফল 0 হওয়া অবৈধ। যদি কম্বাইনারগুলি পাস না করা হয়, ডিফল্ট হল সমস্ত টেবিলের জন্য 'সমষ্টি' ব্যবহার করা। |
---|
পাবলিক স্ট্যাটিক EnqueueTPUEmbeddingRaggedTensorBatch (তৈরি ব্যাপ্তি সুযোগ Iterable < প্রতীক <টি >> sampleSplits, Iterable < প্রতীক <u >> embeddingIndices, Iterable < প্রতীক <ভী >> aggregationWeights, প্রতীক <স্ট্রিং> modeOverride, তালিকা <লং> tableIds, বিকল্প .. । অপশন)
একটি নতুন EnqueueTPUEmbeddingRaggedTensorBatch অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানার পদ্ধতি।
পরামিতি
সুযোগ | বর্তমান সুযোগ |
---|---|
নমুনা স্প্লিট | র্যাঙ্ক 1 টেনসরের একটি তালিকা যা এম্বেডিং_সূচক এবং একত্রিত_ওজন সারিগুলিতে বিভক্ত করার জন্য বিরতি পয়েন্টগুলি নির্দিষ্ট করে৷ এটি embedding_lookup() এ ids.row_splits এর সাথে মিলে যায়, যখন ids একটি RaggedTensor হয়। |
এমবেডিং সূচক | র্যাঙ্ক 1 টেনসরের একটি তালিকা, এম্বেডিং টেবিলের মধ্যে সূচক। এটি embedding_lookup() এর ids.values-এর সাথে মিলে যায়, যখন ids একটি RaggedTensor হয়। |
একত্রিত ওজন | র্যাঙ্ক 1 টেনসরের একটি তালিকা যেখানে প্রতি প্রশিক্ষণের উদাহরণ একত্রিতকরণ ওজন রয়েছে। এটি একটি RaggedTensor-এর মান ক্ষেত্রের সাথে সামঞ্জস্যপূর্ণ, যার সাথে একই row_splits এম্বেডিং_লুকআপ(), যখন ids একটি RaggedTensor হয়। |
মোড ওভাররাইড | একটি স্ট্রিং ইনপুট যা TPUEmbedding কনফিগারেশনে নির্দিষ্ট করা মোডকে ওভাররাইড করে। সমর্থিত মান হল {'unspecified', 'inference', 'training', 'backward_pass_only'}। 'অনির্দিষ্ট'-এ সেট করা হলে, TPUEmbeddingConfiguration-এ সেট করা মোড ব্যবহার করা হয়, অন্যথায় mode_override ব্যবহার করা হয়। |
টেবিল আইডি | সংশ্লিষ্ট ইনপুট খোঁজার জন্য এমবেডিং টেবিলের (TPUEmbedding কনফিগারেশনে TableDescriptor অফসেট) সনাক্তকারী নির্দিষ্ট করে পূর্ণসংখ্যার একটি তালিকা। table_ids[i] ব্যবহার করে ith ইনপুট দেখা হয়। টেবিল_আইডি তালিকার আকার অবশ্যই নমুনা_সূচক, এম্বেডিং_সূচক এবং সমষ্টি_ওজন এর সমান হতে হবে। |
বিকল্প | ঐচ্ছিক বৈশিষ্ট্য মান বহন করে |
রিটার্নস
- EnqueueTPUEmbeddingRaggedTensorBatch এর একটি নতুন উদাহরণ
পাবলিক স্ট্যাটিক EnqueueTPUEmbeddingRaggedTensorBatch.Options deviceOrdinal (দীর্ঘ deviceOrdinal)
পরামিতি
ডিভাইস অর্ডিনাল | টিপিইউ ডিভাইসটি ব্যবহার করতে হবে। যে টাস্কটিতে নোডটি স্থাপন করা হয়েছে তাতে TPU কোরের সংখ্যা থেকে >= 0 এবং কম হওয়া উচিত। |
---|