TPUEmbeddingActivations

lớp cuối cùng công khai TPUEmbeddingActivations

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
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.
Đầu ra < TFloat32 >

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

Giá trị không đổi: "TPUEmbeddingActivations"

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

Đầu ra công khai < TFloat32 > đầu ra ()