EnqueueTPUEmbeddingRaggedTensorBatch

পাবলিক চূড়ান্ত বর্গ EnqueueTPUEmbeddingRaggedTensorBatch

কোডের পোর্টিং সহজ করে যা 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 এবং কম হওয়া উচিত।

পাবলিক স্ট্যাটিক EnqueueTPUEmbeddingRaggedTensorBatch.Options maxSequenceLengths (তালিকা <লং> maxSequenceLengths)

পাবলিক স্ট্যাটিক EnqueueTPUEmbeddingRaggedTensorBatch.Options numFeatures (তালিকা <লং> numFeatures)