একটি অপশন যা একটি 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 | |
স্ট্যাটিক 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 এবং কম হওয়া উচিত। |
---|