একটি অপ যা TPU এ এমবেডিং অ্যাক্টিভেশন গ্রহণ করে।
TPU সিস্টেম TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor) ব্যাচে আর্গুমেন্ট দ্বারা নির্দিষ্ট করা এমবেডিং লুকআপ এবং সমষ্টি সম্পাদন করে। এই সমষ্টির ফলাফলগুলি টেনসরফ্লো গ্রাফে একটি RecvTPUEmbeddingActivations op এর আউটপুট হিসাবে দৃশ্যমান। এই অপশনটি মডেলে নির্দিষ্ট করা প্রতি টেবিলে সক্রিয়করণের একটি টেনসর সমন্বিত একটি তালিকা প্রদান করে। TPU গ্রাফে সর্বাধিক একটি RecvTPUEmbeddingActivations অপশন থাকতে পারে।
পাবলিক পদ্ধতি
স্ট্যাটিক RecvTPUEmbeddingActivations | |
Iterator < প্রতীক <ফ্লোট >> | |
তালিকা < আউটপুট <ফ্লোট >> | আউটপুট () মডেলের প্রতি এম্বেডিং টেবিলে একটি টেনসর সমন্বিত এমবেডিং অ্যাক্টিভেশনের একটি টেনসরলিস্ট। |
উত্তরাধিকারসূত্রে প্রাপ্ত পদ্ধতি
পাবলিক পদ্ধতি
পাবলিক স্ট্যাটিক RecvTPUEmbeddingActivations তৈরি করুন ( ব্যাপ্তি সুযোগ, লং numOutputs, স্ট্রিং কনফিগ)
একটি নতুন RecvTPUEmbeddingActivations অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানা পদ্ধতি।
পরামিতি
সুযোগ | বর্তমান সুযোগ |
---|---|
numOutputs | আউটপুট অ্যাক্টিভেশন টেনসরের সংখ্যা, মডেলটিতে এমবেডিং টেবিলের সংখ্যার সমান। |
কনফিগারেশন | ক্রমিককৃত TPUEmbedding কনফিগারেশন প্রোটো। |
রিটার্নস
- RecvTPUEmbeddingActivations এর একটি নতুন উদাহরণ
প্রকাশ্য তালিকা < আউটপুট <ফ্লোট >> আউটপুট ()
মডেলের প্রতি এম্বেডিং টেবিলে একটি টেনসর সমন্বিত এমবেডিং অ্যাক্টিভেশনের একটি টেনসরলিস্ট।