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 tensor `input`:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
[2, 2]
[1, 1]
[[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
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. |
---|