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.
Các lớp lồng nhau
lớp học | TPUPartitionedOutput.Options | Các thuộc tính tùy chọn cho TPUPartitionedOutput |
Phương pháp công cộng
tĩnh <T> TPUPartitionedOutput <T> | tạo ( Phạm vi phạm vi, đầu vào Toán hạng <T>, Tùy chọn numSplits dài, Tùy chọn... ) Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TPUPartitionedOutput 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 vào được phân vùng phải có hình dạng giống nhau. |
tĩnh TPUPartitionedOutput.Options | phân vùngDim (Phân vùng dàiDim) |
Phương pháp kế thừa
Phương pháp công cộng
tĩnh công khai TPUPartitionedOutput <T> tạo (Phạm vi phạm vi , đầu vào Toán hạng <T>, numSplits dài, Tùy chọn... tùy chọn)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác TPUPartitionedOutput 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. |
tùy chọn | mang các giá trị thuộc tính tùy chọn |
Trả lại
- một phiên bản mới của TPUPartitionedOutput
Danh sách công khai< Đầu ra <T>> đầu ra ()
Danh sách các đầu vào được phân vùng phải có hình dạng giống nhau.
phân vùng tĩnh công khai TPUPartitionedOutput.OptionsDim (Phân vùng dàiDim)
Thông số
phân vùng mờ | Một số nguyên mô tả kích thước nào được phân chia. |
---|