SendTPUEmbeddingGradients
Sử dụng bộ sưu tập để sắp xếp ngăn nắp các trang
Lưu và phân loại nội dung dựa trên lựa chọn ưu tiên của bạn.
Thực hiện cập nhật độ dốc của các bảng nhúng.
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 kế thừa
Từ lớp java.lang.Object boolean | bằng (Đối tượng arg0) |
Lớp cuối cùng<?> | getClass () |
int | Mã Băm () |
khoảng trống cuối cùng | thông báo () |
khoảng trống cuối cùng | thông báoTất cả () |
Sợi dây | toString () |
khoảng trống cuối cùng | chờ đã (arg0 dài, int arg1) |
khoảng trống cuối cùng | chờ đã (arg0 dài) |
khoảng trống cuối cùng | Chờ đợi () |
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: "SendTPUEmbeddingGradients"
Phương pháp công khai
Phương thức xuất xưởng để tạo một lớp bao bọc hoạt động SendTPUEmbeddingGradients mới.
Thông số
phạm vi | phạm vi hiện tại |
---|
đầu vào | Một TensorList gồm các gradient dùng để cập nhật các bảng nhúng. Đối số này có cùng độ dài và hình dạng với giá trị trả về của RecvTPUEmbeddingActivations, nhưng chứa độ dốc của sự mất mát của mô hình đối với các kích hoạt nhúng. Các bảng nhúng được cập nhật từ các gradient này thông qua trình tối ưu hóa được chỉ định trong cấu hình nhúng TPU được cung cấp cho tpu.initialize_system. |
---|
tỷ lệ học tập | Danh sách Tensor gồm các giá trị vô hướng float32, một cho mỗi thẻ tốc độ học động: xem nhận xét trong //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Nhiều bảng có thể chia sẻ cùng một thẻ tốc độ học tập động như được chỉ định trong cấu hình. Nếu tốc độ học của tất cả các bảng là không đổi thì danh sách này sẽ trống. |
---|
cấu hình | Proto Cấu hình TPUEmbedding được tuần tự hóa. |
---|
Trả lại
- một phiên bản mới của SendTPUEmbeddingGradients
Trừ phi có lưu ý khác, nội dung của trang này được cấp phép theo Giấy phép ghi nhận tác giả 4.0 của Creative Commons và các mẫu mã lập trình được cấp phép theo Giấy phép Apache 2.0. Để biết thông tin chi tiết, vui lòng tham khảo Chính sách trang web của Google Developers. Java là nhãn hiệu đã đăng ký của Oracle và/hoặc các đơn vị liên kết với Oracle.
Cập nhật lần gần đây nhất: 2025-07-26 UTC.
[null,null,["Cập nhật lần gần đây nhất: 2025-07-26 UTC."],[],[],null,["# SendTPUEmbeddingGradients\n\npublic final class **SendTPUEmbeddingGradients** \nPerforms gradient updates of embedding tables. \n\n### Constants\n\n|--------|---------------------------------------------------------------------------------------|---------------------------------------------------------|\n| String | [OP_NAME](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients#OP_NAME) | The name of this op, as known by TensorFlow core engine |\n\n### Public Methods\n\n|--------------------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| static [SendTPUEmbeddingGradients](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients) | [create](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients#create(org.tensorflow.op.Scope, java.lang.Iterable\u003corg.tensorflow.Operand\u003corg.tensorflow.types.TFloat32\u003e\u003e, java.lang.Iterable\u003corg.tensorflow.Operand\u003corg.tensorflow.types.TFloat32\u003e\u003e, java.lang.String))([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e inputs, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e learningRates, String config) Factory method to create a class wrapping a new SendTPUEmbeddingGradients operation. |\n\n### Inherited Methods\n\nFrom class [org.tensorflow.op.RawOp](/jvm/api_docs/java/org/tensorflow/op/RawOp) \n\n|----------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| final boolean | [equals](/jvm/api_docs/java/org/tensorflow/op/RawOp#equals(java.lang.Object))(Object obj) |\n| final int | [hashCode](/jvm/api_docs/java/org/tensorflow/op/RawOp#hashCode())() |\n| [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/RawOp#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n| final String | [toString](/jvm/api_docs/java/org/tensorflow/op/RawOp#toString())() |\n\nFrom class java.lang.Object \n\n|------------------|---------------------------|\n| boolean | equals(Object arg0) |\n| final Class\\\u003c?\\\u003e | getClass() |\n| int | hashCode() |\n| final void | notify() |\n| final void | notifyAll() |\n| String | toString() |\n| final void | wait(long arg0, int arg1) |\n| final void | wait(long arg0) |\n| final void | wait() |\n\nFrom interface [org.tensorflow.op.Op](/jvm/api_docs/java/org/tensorflow/op/Op) \n\n|-----------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract [ExecutionEnvironment](/jvm/api_docs/java/org/tensorflow/ExecutionEnvironment) | [env](/jvm/api_docs/java/org/tensorflow/op/Op#env())() Return the execution environment this op was created in. |\n| abstract [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/Op#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n\nConstants\n---------\n\n#### public static final String\n**OP_NAME**\n\nThe name of this op, as known by TensorFlow core engine \nConstant Value: \"SendTPUEmbeddingGradients\"\n\nPublic Methods\n--------------\n\n#### public static [SendTPUEmbeddingGradients](/jvm/api_docs/java/org/tensorflow/op/tpu/SendTPUEmbeddingGradients)\n**create**\n([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e inputs, Iterable\\\u003c[Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TFloat32](/jvm/api_docs/java/org/tensorflow/types/TFloat32)\\\u003e\\\u003e learningRates, String config)\n\nFactory method to create a class wrapping a new SendTPUEmbeddingGradients operation. \n\n##### Parameters\n\n| scope | current scope |\n| inputs | A TensorList of gradients with which to update embedding tables. This argument has the same length and shapes as the return value of RecvTPUEmbeddingActivations, but contains gradients of the model's loss with respect to the embedding activations. The embedding tables are updated from these gradients via the optimizer specified in the TPU embedding configuration given to tpu.initialize_system. |\n| learningRates | A TensorList of float32 scalars, one for each dynamic learning rate tag: see the comments in //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto. Multiple tables can share the same dynamic learning rate tag as specified in the configuration. If the learning rates for all tables are constant, this list should be empty. |\n| config | Serialized TPUEmbeddingConfiguration proto. |\n|---------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n\n##### Returns\n\n- a new instance of SendTPUEmbeddingGradients"]]