XlaRecvTPUEmbeddingActivations

সর্বজনীন চূড়ান্ত ক্লাস XlaRecvTPUEmbeddingActivations

একটি অপ যা TPU এ এমবেডিং অ্যাক্টিভেশন গ্রহণ করে।

TPU সিস্টেম এমবেডিং লুকআপ এবং একত্রিতকরণ সম্পাদন করে। এই সমষ্টির ফলাফলগুলি একটি XlaRecvTPUEmbeddingActivations Op-এর আউটপুট হিসাবে টেনসরফ্লো গ্রাফে দৃশ্যমান। এই অপশনটি মডেলে নির্দিষ্ট করা প্রতি টেবিলে সক্রিয়করণের একটি টেনসর সমন্বিত একটি তালিকা প্রদান করে।

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

স্ট্যাটিক XlaRecvTPUEmbedding সক্রিয়করণ
তৈরি করুন ( স্কোপ স্কোপ, অপারেন্ড <?> deduplicationData, লং numTables, String config)
একটি নতুন XlaRecvTPUEmbeddingActivations অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানা পদ্ধতি।
ইটারেটর< অপারেন্ড <ফ্লোট>>
পুনরাবৃত্তিকারী ()
তালিকা< আউটপুট <ফ্লোট>>
আউটপুট ()
মডেলের প্রতি এম্বেডিং টেবিলে একটি টেনসর সমন্বিত এমবেডিং অ্যাক্টিভেশনের একটি টেনসরলিস্ট।

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

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

পাবলিক স্ট্যাটিক XlaRecvTPUEmbeddingActivations তৈরি করে ( Scope scope, Operand <?> deduplicationData, Long numTables, String config)

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

পরামিতি
সুযোগ বর্তমান সুযোগ
deduplicationData টাইপ=DT_VARIANT সহ একটি টেনসর যাতে ডিডপ্লিকেশন ডেটা থাকে। টেনসর হল একটি XLA নেস্টেড টিপল যাতে N উপাদান থাকে (যেখানে N হল প্রতি TPU চিপে টেনসর কোরের এমবেডিংয়ের সংখ্যার অনুপাত)। নেস্টেড টিপলের প্রতিটি উপাদান হল র্যাঙ্ক 1 টেনসরের একটি টিপল। প্রতিটি টেনসরে হয় টেনসরকোরে এম্বেড করার জন্য সূচক (DT_UINT32) বা এমবেডিং লুকআপ অপারেশনের আউটপুটে প্রয়োগ করার জন্য ওজন (DT_FLOAT) থাকে।
সংখ্যা টেবিল আউটপুট অ্যাক্টিভেশন টেনসরের সংখ্যা। টিপিইউ এম্বেডিং কনফিগারে যদি ফিচার ডিসক্রিপ্টর উপস্থিত থাকে, তবে এটি বৈশিষ্ট্যের সংখ্যার সমান অন্যথায় মডেলের এম্বেডিং টেবিলের সংখ্যার সমান।
কনফিগারেশন ক্রমিককৃত TPUEmbedding কনফিগারেশন প্রোটো।
রিটার্নস
  • XlaRecvTPUEmbeddingActivations এর একটি নতুন উদাহরণ

পাবলিক ইটারেটর< অপারেন্ড <ফ্লোট>> পুনরাবৃত্তিকারী ()

সর্বজনীন তালিকা< আউটপুট <ফ্লোট>> আউটপুট ()

মডেলের প্রতি এম্বেডিং টেবিলে একটি টেনসর সমন্বিত এমবেডিং অ্যাক্টিভেশনের একটি টেনসরলিস্ট।