lớp cuối cùng công khai TPUPartitionedOutputV2
Một op phân tách một tensor được XLA phân chia thành một danh sách được phân vùng
đầu ra bên ngoài tính toán XLA. Hỗ trợ phân mảnh ND.
Phương pháp công khai
| tĩnh <T> TPUPartitionedOutputV2 <T> | tạo ( Phạm vi phạm vi, đầu vào Toán hạng <T>, numSplits dài, Danh sách<Long> phân vùngDims) Phương thức gốc để tạo một lớp bao bọc một thao tác TPUPartitionedOutputV2 mới. |
| Trình vòng lặp< Toán hạng <T>> | trình lặp () |
| Danh sách< Đầu ra <T>> | đầu ra () Danh sách các đầu ra được phân vùng có hình dạng giống nhau. |
Phương pháp kế thừa
Phương pháp công khai
tĩnh công khai TPUPartitionedOutputV2 <T> tạo (Phạm vi phạm vi , đầu vào Toán hạng <T>, numSplits dài, Phân vùng danh sách<Long>Dims)
Phương thức gốc để tạo một lớp bao bọc một thao tác TPUPartitionedOutputV2 mới.
Thông số
| phạm vi | phạm vi hiện tại |
|---|---|
| đầu vào | Một tensor đại diện cho hình dạng đầy đủ của các tensor được phân vùng. |
| phân vùngDims | Danh sách các số nguyên mô tả cách mỗi chiều được phân chia. Khoảng trống cho biết đầu vào được sao chép. |
Trả lại
- một phiên bản mới của TPUPartitionedOutputV2
Danh sách công khai< Đầu ra <T>> đầu ra ()
Danh sách các đầu ra được phân vùng có hình dạng giống nhau.