XlaSplitND

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

Chia tensor đầu vào theo tất cả các chiều.

Một op cắt tensor đầu vào dựa trên thuộc tính num_splits đã cho, đệm các lát tùy ý và trả về các lát. Các lát cắt được trả về theo thứ tự hàng lớn.

Hoạt động này có thể được tạo thông qua cầu nối TPU.

Ví dụ: với `input` tensor: 

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
`num_splits`: 
[2, 2]
và `paddings`: 
[1, 1]
thì `outputs` dự kiến ​​là: 
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]

Các lớp lồng nhau

lớp học XlaSplitND.Options Thuộc tính tùy chọn cho XlaSplitND

Phương pháp công khai

tĩnh <T> XlaSplitND <T>
tạo ( Phạm vi phạm vi, đầu vào Toán hạng <T>, Long N, Danh sách<Long> numSplits, 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 XlaSplitND 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 ()
XlaSplitND.Options tĩnh
phần đệm (Danh sách phần đệm <Long>)

Phương pháp kế thừa

Phương pháp công khai

public static XlaSplitND <T> tạo (Phạm vi phạm vi, đầu vào Toán hạng <T>, Long N, Danh sách<Long> numSplits, 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 XlaSplitND mới.

Thông số
phạm vi phạm vi hiện tại
đầu vào Tenxor đầu vào để phân chia theo tất cả các chiều. } out_arg { tên: mô tả "đầu ra": <
sốSplits Số cách chia theo chiều. Kích thước hình dạng phải được chia đều.
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 XlaSplitND

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

Danh sách công khai< Đầu ra <T>> đầu ra ()

phần đệm XlaSplitND.Options tĩnh công khai (Phần đệm danh sách <Long>)

Thông số
phần đệm Danh sách tùy chọn các phần đệm bên phải theo chiều của tensor đầu vào sẽ được áp dụng trước khi phân tách. Điều này có thể được sử dụng để làm cho một kích thước có thể chia đều.