একটি অপ যা TPU এ এমবেডিং অ্যাক্টিভেশন গ্রহণ করে।
TPU সিস্টেম এমবেডিং লুকআপ এবং একত্রিতকরণ সম্পাদন করে। এই সমষ্টির ফলাফলগুলি একটি XlaRecvTPUEmbeddingActivations Op-এর আউটপুট হিসাবে টেনসরফ্লো গ্রাফে দৃশ্যমান। এই অপশনটি মডেলে নির্দিষ্ট করা প্রতি টেবিলে সক্রিয়করণের একটি টেনসর সমন্বিত একটি তালিকা প্রদান করে।
পাবলিক পদ্ধতি
স্ট্যাটিক XlaRecvTPUEmbedding সক্রিয়করণ | |
ইটারেটর< অপারেন্ড <ফ্লোট>> | |
তালিকা< আউটপুট <ফ্লোট>> | আউটপুট () মডেলের প্রতি এম্বেডিং টেবিলে একটি টেনসর সমন্বিত এমবেডিং অ্যাক্টিভেশনের একটি টেনসরলিস্ট। |
উত্তরাধিকারসূত্রে প্রাপ্ত পদ্ধতি
পাবলিক পদ্ধতি
পাবলিক স্ট্যাটিক XlaRecvTPUEmbeddingActivations তৈরি করে ( Scope scope, Operand <?> deduplicationData, Long numTables, String config)
একটি নতুন XlaRecvTPUEmbeddingActivations অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানা পদ্ধতি।
পরামিতি
সুযোগ | বর্তমান সুযোগ |
---|---|
deduplicationData | টাইপ=DT_VARIANT সহ একটি টেনসর যাতে ডিডপ্লিকেশন ডেটা থাকে। টেনসর হল একটি XLA নেস্টেড টিপল যাতে N উপাদান থাকে (যেখানে N হল প্রতি TPU চিপে টেনসর কোরের এমবেডিংয়ের সংখ্যার অনুপাত)। নেস্টেড টিপলের প্রতিটি উপাদান হল র্যাঙ্ক 1 টেনসরের একটি টিপল। প্রতিটি টেনসরে হয় টেনসরকোরে এম্বেড করার জন্য সূচক (DT_UINT32) বা এমবেডিং লুকআপ অপারেশনের আউটপুটে প্রয়োগ করার জন্য ওজন (DT_FLOAT) থাকে। |
সংখ্যা টেবিল | আউটপুট অ্যাক্টিভেশন টেনসরের সংখ্যা। টিপিইউ এম্বেডিং কনফিগারে যদি ফিচার ডিসক্রিপ্টর উপস্থিত থাকে, তবে এটি বৈশিষ্ট্যের সংখ্যার সমান অন্যথায় মডেলের এম্বেডিং টেবিলের সংখ্যার সমান। |
কনফিগারেশন | ক্রমিককৃত TPUEmbedding কনফিগারেশন প্রোটো। |
রিটার্নস
- XlaRecvTPUEmbeddingActivations এর একটি নতুন উদাহরণ
সর্বজনীন তালিকা< আউটপুট <ফ্লোট>> আউটপুট ()
মডেলের প্রতি এম্বেডিং টেবিলে একটি টেনসর সমন্বিত এমবেডিং অ্যাক্টিভেশনের একটি টেনসরলিস্ট।