XlaSparseDenseMatmul

सार्वजनिक अंतिम कक्षा XlaSparseDenseMatmul

सार्वजनिक तरीके

आउटपुट <फ्लोट>
स्थिर XlaSparseDenseMatmul
बनाएं ( स्कोप स्कोप, ऑपरेंड <इंटेगर> पंक्ति आईडी, ऑपरेंड <?> कॉलआईड्स, ऑपरेंड <फ्लोट> मान, ऑपरेंड <?> ऑफसेट, ऑपरेंड <फ्लोट> एम्बेडिंगटेबल, लॉन्ग मैक्सआईडीएसपरपार्टिशन, लॉन्ग मैक्सयूनीकआईडीपरपार्टिशन, लॉन्ग इनपुटसाइज)
एक नए XlaSparseDenseMatmul ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
आउटपुट <पूर्णांक>
आउटपुट <पूर्णांक>
आउटपुट <फ्लोट>
आउटपुट <पूर्णांक>

विरासत में मिली विधियाँ

सार्वजनिक तरीके

सार्वजनिक आउटपुट <फ्लोट> सक्रियण ()

सार्वजनिक स्थैतिक XlaSparseDenseMatmul बनाएं ( स्कोप स्कोप, ऑपरेंड <पूर्णांक> rowIds, ऑपरेंड <?> colIds, ऑपरेंड <फ्लोट> मान, ऑपरेंड <?> ऑफसेट, ऑपरेंड <फ्लोट> एम्बेडिंगटेबल, लंबी maxIdsPerPartition, लंबी maxUniqueIdsPerPartition, लंबी इनपुट आकार)

एक नए XlaSparseDenseMatmul ऑपरेशन को लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।

पैरामीटर
दायरा वर्तमान दायरा
रिटर्न
  • XlaSparseDenseMatmul का एक नया उदाहरण

सार्वजनिक आउटपुट <पूर्णांक> rowPointers ()

सार्वजनिक आउटपुट <पूर्णांक> sortedEmbeddingIds ()

सार्वजनिक आउटपुट <फ्लोट> सॉर्टेडगेन्स ()

सार्वजनिक आउटपुट <Integer> sortedSampleIds ()