TPUPartitionedOutput

lớp cuối cùng công khai TPUPartitionedOutput

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

public Iterator< Toán hạng <T>> iterator ()

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.