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