NcclAllReduce
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.
Xuất ra một tensor chứa mức giảm trên tất cả các tensor đầu vào.
Xuất ra một tensor chứa mức rút gọn trên tất cả các tensor đầu vào được chuyển đến các op trong cùng một `shared_name.
Biểu đồ phải được xây dựng để nếu một op chạy với giá trị Shared_name `c`, thì các op `num_devices` sẽ chạy với giá trị Shared_name `c`. Nếu không làm như vậy sẽ khiến việc thực thi biểu đồ không hoàn thành.
đầu vào: đầu vào của dữ liệu giảm: giá trị giảm trên tất cả các thiết bị `num_devices`. giảm: hoạt động giảm để thực hiện. num_devices: Số lượng thiết bị tham gia đợt giảm này. Shared_name: Mã định danh được chia sẻ giữa các hoạt động có cùng mức giảm.
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 <T> | asOutput () Trả về tay cầm tượng trưng của tensor. |
tĩnh <T mở rộng TNumber > NcclAllReduce <T> | tạo ( Phạm vi phạm vi, đầu vào Toán hạng <T>, Giảm chuỗi, Số thiết bị dài, Tên chia sẻ chuỗi) Phương thức xuất xưởng để tạo một lớp bao bọc thao tác NcclAllReduce mới. |
Đầu ra <T> | |
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 () |
Từ giao diện org.tensorflow.Operand Đầu ra trừu tượng <T> | asOutput () Trả về tay cầm tượng trưng của tensor. |
trừu tượng T | asTensor () Trả về tensor ở toán hạng này. |
hình dạng trừu tượng | hình dạng () Trả về hình dạng (có thể đã biết một phần) của tensor được tham chiếu bởi Output của toán hạng này. |
Lớp trừu tượng<T> | kiểu () Trả về kiểu tensor của toán hạng này |
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: "NcclAllReduce"
Phương pháp công khai
Đầu ra công khai <T> 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.
Phương thức xuất xưởng để tạo một lớp bao bọc thao tác NcclAllReduce mới.
Trả lại
- một phiên bản mới của NcclAllReduce
Dữ liệu đầu ra công khai <T> ()
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-25 UTC.
[null,null,["Cập nhật lần gần đây nhất: 2025-07-25 UTC."],[],[],null,["# NcclAllReduce\n\npublic final class **NcclAllReduce** \nOutputs a tensor containing the reduction across all input tensors.\n\n\nOutputs a tensor containing the reduction across all input tensors passed to ops\nwithin the same \\`shared_name.\n\n\nThe graph should be constructed so if one op runs with shared_name value \\`c\\`,\nthen \\`num_devices\\` ops will run with shared_name value \\`c\\`. Failure to do so\nwill cause the graph execution to fail to complete.\n\n\ninput: the input to the reduction\ndata: the value of the reduction across all \\`num_devices\\` devices.\nreduction: the reduction operation to perform.\nnum_devices: The number of devices participating in this reduction.\nshared_name: Identifier that shared between ops of the same reduction.\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n### Constants\n\n|--------|----------------------------------------------------------------------------------|---------------------------------------------------------|\n| String | [OP_NAME](/jvm/api_docs/java/org/tensorflow/op/distribute/NcclAllReduce#OP_NAME) | The name of this op, as known by TensorFlow core engine |\n\n### Public Methods\n\n|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e | [asOutput](/jvm/api_docs/java/org/tensorflow/op/distribute/NcclAllReduce#asOutput())() Returns the symbolic handle of the tensor. |\n| static \\\u003cT extends [TNumber](/jvm/api_docs/java/org/tensorflow/types/family/TNumber)\\\u003e [NcclAllReduce](/jvm/api_docs/java/org/tensorflow/op/distribute/NcclAllReduce)\\\u003cT\\\u003e | [create](/jvm/api_docs/java/org/tensorflow/op/distribute/NcclAllReduce#create(org.tensorflow.op.Scope, org.tensorflow.Operand\u003cT\u003e, java.lang.String, java.lang.Long, java.lang.String))([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e input, String reduction, Long numDevices, String sharedName) Factory method to create a class wrapping a new NcclAllReduce operation. |\n| [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e | [data](/jvm/api_docs/java/org/tensorflow/op/distribute/NcclAllReduce#data())() |\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\nFrom interface [org.tensorflow.Operand](/jvm/api_docs/java/org/tensorflow/Operand) \n\n|-------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e | [asOutput](/jvm/api_docs/java/org/tensorflow/Operand#asOutput())() Returns the symbolic handle of the tensor. |\n| abstract T | [asTensor](/jvm/api_docs/java/org/tensorflow/Operand#asTensor())() Returns the tensor at this operand. |\n| abstract [Shape](/jvm/api_docs/java/org/tensorflow/ndarray/Shape) | [shape](/jvm/api_docs/java/org/tensorflow/Operand#shape())() Returns the (possibly partially known) shape of the tensor referred to by the [Output](/jvm/api_docs/java/org/tensorflow/Output) of this operand. |\n| abstract Class\\\u003cT\\\u003e | [type](/jvm/api_docs/java/org/tensorflow/Operand#type())() Returns the tensor type of this operand |\n\nFrom interface [org.tensorflow.ndarray.Shaped](/jvm/api_docs/java/org/tensorflow/ndarray/Shaped) \n\n|-------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract int | [rank](/jvm/api_docs/java/org/tensorflow/ndarray/Shaped#rank())() |\n| abstract [Shape](/jvm/api_docs/java/org/tensorflow/ndarray/Shape) | [shape](/jvm/api_docs/java/org/tensorflow/ndarray/Shaped#shape())() |\n| abstract long | [size](/jvm/api_docs/java/org/tensorflow/ndarray/Shaped#size())() Computes and returns the total size of this container, in number of values. |\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: \"NcclAllReduce\"\n\nPublic Methods\n--------------\n\n#### public [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e\n**asOutput**\n()\n\nReturns the symbolic handle of the tensor.\n\nInputs to TensorFlow operations are outputs of another TensorFlow operation. This method is\nused to obtain a symbolic handle that represents the computation of the input.\n\n\u003cbr /\u003e\n\n#### public static [NcclAllReduce](/jvm/api_docs/java/org/tensorflow/op/distribute/NcclAllReduce)\\\u003cT\\\u003e\n**create**\n([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e input, String reduction, Long numDevices, String sharedName)\n\nFactory method to create a class wrapping a new NcclAllReduce operation. \n\n##### Parameters\n\n| scope | current scope |\n|-------|---------------|\n\n##### Returns\n\n- a new instance of NcclAllReduce \n\n#### public [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e\n**data**\n()\n\n\u003cbr /\u003e"]]