Tách một tensor thành các tensor `num_split` dọc theo một chiều.
Phương pháp công khai
tĩnh <T, U mở rộng Số> SplitV <T> | |
Trình vòng lặp< Toán hạng <T>> | trình lặp () |
Danh sách< Đầu ra <T>> | đầu ra () Các tensor có hình dạng khớp với hình dạng của `value` ngoại trừ dọc theo `axis`, trong đó kích thước của chúng là `size_splits[i]`. |
Phương pháp kế thừa
Phương pháp công khai
công khai tĩnh SplitV <T> tạo ( Phạm vi phạm vi, giá trị Toán hạng <T>, Toán hạng <U> sizeSplits, Trục toán hạng <Số nguyên>, numSplit dài)
Phương thức xuất xưởng để tạo một lớp bao bọc một thao tác SplitV mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
giá trị | Các tensor để phân chia. |
kích thướcSplits | danh sách chứa kích thước của từng tensor đầu ra dọc theo chiều phân chia. Phải tính tổng theo thứ nguyên của giá trị dọc theo Split_dim. Có thể chứa một -1 cho biết thứ nguyên đó sẽ được suy ra. |
trục | 0-D. Kích thước để phân chia. Phải nằm trong phạm vi `[-rank(value), Rank(value))`. |
Trả lại
- một phiên bản mới của SplitV
Danh sách công khai< Đầu ra <T>> đầu ra ()
Các tensor có hình dạng khớp với hình dạng của `value` ngoại trừ dọc theo `axis`, trong đó kích thước của chúng là `size_splits[i]`.