EnqueueTPUEmbeddingSparseBatch

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

একটি অপশন যা একটি SparseTensor থেকে TPUEmbedding ইনপুট সূচক সারিবদ্ধ করে।

এই অপটি কোডের পোর্টিং সহজ করে যা embedding_lookup_sparse() ব্যবহার করে, যদিও SparseTensor আর্গুমেন্টের কিছু Python preprocessing to embedding_lookup_sparse() এই Op-এ আর্গুমেন্ট তৈরি করতে হবে, যেহেতু শুধুমাত্র একটি EnqueueTPUEmbeddingSparseBatch Op প্রশিক্ষণের জন্য অনুমোদিত।

তিনটি ইনপুট তালিকার সংশ্লিষ্ট অবস্থানে থাকা টেনসরগুলির অবশ্যই একই আকৃতি থাকতে হবে, যেমন র‍্যাঙ্ক 1 এবং dim_size() সংশ্লিষ্ট table_id দ্বারা বর্ণিত টেবিলে মোট লুকআপ সংখ্যার সমান।

নেস্টেড ক্লাস

ক্লাস EnqueueTPUEmbeddingSparseBatch.Options জন্য ঐচ্ছিক বৈশিষ্ট্য EnqueueTPUEmbeddingSparseBatch

পাবলিক পদ্ধতি

স্ট্যাটিক EnqueueTPUEmbeddingSparseBatch.Options
combiners (তালিকা <স্ট্রিং> combiners)
স্ট্যাটিক <টি প্রসারিত নম্বর, ইউ নম্বর প্রসারিত, ভি প্রসারিত number> EnqueueTPUEmbeddingSparseBatch
তৈরি ( ব্যাপ্তি সুযোগ Iterable < প্রতীক <টি >> sampleIndices, Iterable < প্রতীক <u >> embeddingIndices, Iterable < প্রতীক <ভী >> aggregationWeights, প্রতীক <স্ট্রিং> modeOverride, বিকল্পসমূহ ... অপশন)
একটি নতুন EnqueueTPUEmbeddingSparseBatch অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানার পদ্ধতি।
স্ট্যাটিক EnqueueTPUEmbeddingSparseBatch.Options
deviceOrdinal (দীর্ঘ deviceOrdinal)

উত্তরাধিকারসূত্রে প্রাপ্ত পদ্ধতি

পাবলিক পদ্ধতি

পাবলিক স্ট্যাটিক EnqueueTPUEmbeddingSparseBatch.Options combiners (তালিকা <স্ট্রিং> combiners)

পরামিতি
সংযোজক স্ট্রিং স্কেলারগুলির একটি তালিকা, প্রতিটি এম্বেডিং টেবিলের জন্য একটি যা নির্দিষ্ট করে কিভাবে ওজনযুক্ত যোগফলের পরে এমবেডিং অ্যাক্টিভেশনগুলিকে স্বাভাবিক করা যায়। সমর্থিত কম্বাইনারগুলি হল 'মান', 'সমর্থন', বা 'sqrtn'। 'মান'-এর জন্য ওজনের যোগফল 0 বা 'sqrtn'-এর জন্য বর্গ ওজনের যোগফল 0 হওয়া অবৈধ। যদি কম্বাইনারগুলি পাস না করা হয়, ডিফল্ট হল সমস্ত টেবিলের জন্য 'সমষ্টি' ব্যবহার করা।

পাবলিক স্ট্যাটিক EnqueueTPUEmbeddingSparseBatch তৈরি করুন ( ব্যাপ্তি সুযোগ Iterable < প্রতীক <টি >> sampleIndices, Iterable < প্রতীক <u >> embeddingIndices, Iterable < প্রতীক <ভী >> aggregationWeights, প্রতীক <স্ট্রিং> modeOverride, বিকল্পসমূহ ... অপশন)

একটি নতুন EnqueueTPUEmbeddingSparseBatch অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানার পদ্ধতি।

পরামিতি
সুযোগ বর্তমান সুযোগ
নমুনা সূচক র‍্যাঙ্ক 1 টেনসরের একটি তালিকা যা প্রশিক্ষণের উদাহরণ এবং বৈশিষ্ট্য উল্লেখ করে যার সাথে সংশ্লিষ্ট এমবেডিং_সূচক এবং সমষ্টি_ওজন মানগুলি অন্তর্গত। নমুনা_সূচক[i] অবশ্যই b * nf + f এর সমান হবে, যেখানে nf হল সংশ্লিষ্ট টেবিলের বৈশিষ্ট্যের সংখ্যা, f হল [0, nf-এ, এবং b হল [0, ব্যাচের আকার)।
এমবেডিং সূচক র্যাঙ্ক 1 টেনসরের একটি তালিকা, এম্বেডিং টেবিলের মধ্যে সূচক।
একত্রিত ওজন প্রতি নমুনা সহ র্যাঙ্ক 1 টেনসরের একটি তালিকা -- যেমন প্রতি (প্রশিক্ষণের উদাহরণ, বৈশিষ্ট্য) -- একত্রিত ওজন।
মোড ওভাররাইড একটি স্ট্রিং ইনপুট যা TPUEmbedding কনফিগারেশনে নির্দিষ্ট করা মোডকে ওভাররাইড করে। সমর্থিত মান হল {'unspecified', 'inference', 'training', 'backward_pass_only'}। 'অনির্দিষ্ট'-এ সেট করা হলে, TPUEmbeddingConfiguration-এ সেট করা মোড ব্যবহার করা হয়, অন্যথায় mode_override ব্যবহার করা হয়।
বিকল্প ঐচ্ছিক বৈশিষ্ট্য মান বহন করে
রিটার্নস
  • EnqueueTPUEmbeddingSparseBatch এর একটি নতুন উদাহরণ

পাবলিক স্ট্যাটিক EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (দীর্ঘ deviceOrdinal)

পরামিতি
ডিভাইস অর্ডিনাল টিপিইউ ডিভাইসটি ব্যবহার করতে হবে। যে টাস্কটিতে নোডটি স্থাপন করা হয়েছে তাতে TPU কোরের সংখ্যা থেকে >= 0 এবং কম হওয়া উচিত।