Kết hợp tensor đầu vào trên tất cả các chiều.
Một op hợp nhất sẽ cắt các tensor đầu vào dựa trên thuộc tính num_splits đã cho, loại bỏ các phần đệm tùy ý và ghi tensor đã hợp nhất mà không có phần đệm vào biến tài nguyê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],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
`num_splits`: [2, 2]
và `phần đệm`: [1, 1]
`đầu ra` dự kiến là: [[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
Các lớp lồng nhau
lớp học | GánVariableXlaConcatND.Options | Thuộc tính tùy chọn cho AssignVariableXlaConcatND |
Phương pháp công khai
tĩnh <T> GánVariableXlaConcatND | tạo (Phạm vi phạm vi, tài nguyên Toán hạng <?>, Đầu vào Iterable< Toán hạng <T>>, Danh sách<Long> numConcats, Tùy chọn... tùy chọn) Phương thức gốc để tạo một lớp bao bọc một thao tác GánVariableXlaConcatND mới. |
tĩnh GánVariableXlaConcatND.Options | phần đệm (Danh sách phần đệm <Long>) |
Phương pháp kế thừa
Phương pháp công khai
công khai tĩnh GánVariableXlaConcatND tạo (Phạm vi phạm vi, tài nguyên Toán hạng <?>, Đầu vào Iterable< Toán hạng <T>>, Danh sách<Long> numConcats, Tùy chọn... tùy chọn)
Phương thức gốc để tạo một lớp bao bọc một thao tác GánVariableXlaConcatND mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
tài nguyên | Biến tài nguyên cho các tensor đầu vào được nối trên tất cả các chiều. } in_arg { tên: mô tả "đầu vào": < |
sốConcats | Số cách hợp nhất cho mỗi thứ nguyên. |
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 GánVariableXlaConcatND
phần đệm tĩnh công khai GánVariableXlaConcatND.Options (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 cho mỗi chiều để tách khỏi tensor được hợp nhất cuối cùng. Các phần đệm này không được vượt quá kích thước kích thước của kết quả được hợp nhất trước khi loại bỏ phần đệm. |
---|