TPUNhúngKích hoạt

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.

Phương pháp công khai

Đầu ra <Float>
asOutput ()
Trả về phần điều khiển tượng trưng của một tenxơ.
TPUEmbeddingKích hoạt tĩnh
tạo (Phạm vi phạm vi , Toán hạng <Float> embeddingVariable, Toán hạng <Float> cắt látActivations, Id bảng dài, Id tra cứu dài)
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 <Float>
đầu ra ()

Phương pháp kế thừa

Phương pháp công khai

Đầu ra công khai <Float> asOutput ()

Trả về phần điều khiển tượng trưng của một tenxơ.

Đầ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.

Công khai tĩnh TPUEmbeddingActivations tạo (Phạm vi phạm vi , Toán hạng <Float> embeddingVariable, Toán hạng <Float> cắt látActivations, Id bảng dài, Id tra cứu dài)

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 đầu ra <Float> ()