Một cơ hội cho phép phân biệt các vật liệu nhúng TPU.
Hoạt động này chỉ đơn giản trả về đầu vào đầu tiên của nó, được cho là đã được cắt từ các Tensors được trả về bởi TPUEmbeddingDequeueActivations. Sự hiện diện của op này và đối số đầu tiên của nó là Biến có thể huấn luyện, cho phép tự động phân biệt các biểu đồ có chứa phần nhúng thông qua thư viện Python nhúng TPU.
Hằng số
| Sợi dây | OP_NAME | Tên của op này, được biết đến bởi công cụ lõi TensorFlow |
Phương pháp công khai
| Đầu ra < TFloat32 > | asOutput () Trả về tay cầm tượng trưng của tensor. |
| TPUEmbeddingKích hoạt tĩnh | |
| Đầu ra < TFloat32 > | đầu ra () |
Phương pháp kế thừa
Hằng số
Chuỗi cuối cùng tĩnh công khai OP_NAME
Tên của op này, được biết đến bởi công cụ lõi TensorFlow
Phương pháp công khai
Đầu ra công khai < TFloat32 > asOutput ()
Trả về tay cầm tượng trưng của tensor.
Đầu vào của các hoạt động TensorFlow là đầu ra của một hoạt động TensorFlow khác. Phương pháp này được sử dụng để thu được một thẻ điều khiển mang tính biểu tượng đại diện cho việc tính toán đầu vào.
tĩnh công khai TPUEmbeddingActivations tạo (Phạm vi phạm vi , Toán hạng < TFloat32 > embeddingVariable, Toán hạng < TFloat32 > sliceActivations, Long tableId, Long lookupId)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TPUEmbeddingActivations mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| nhúngBiến | Một biến có thể đào tạo, cho phép người tối ưu hóa tìm thấy op này. |
| cắt látKích hoạt | Việc nhúng các kích hoạt Tensor để trả về. |
| id bảng | Id của bảng trong cấu hình lớp nhúng mà từ đó các kích hoạt này được tính toán. |
| tra cứuId | Mã định danh của tập hợp các chỉ số nhúng đã tạo ra các kích hoạt này. |
Trả lại
- một phiên bản mới của TPUEmbeddingActivations