Chia 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> Tách <T> | |
Trình vòng lặp< Toán hạng <T>> | trình lặp () |
Danh sách< Đầu ra <T>> | đầu ra () Chúng là các tensor có hình dạng giống hệt nhau, 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à `values.shape[split_dim] / num_split`. |
Phương pháp kế thừa
Phương pháp công khai
tạo phân tách tĩnh công khai <T> (Phạm vi phạm vi , trục Toán hạng <Số nguyên>, Giá trị toán hạng <T>, 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 Phân chia mới.
Thông số
phạm vi | phạm vi hiện tại |
---|---|
trục | 0-D. Kích thước để phân chia. Phải nằm trong phạm vi `[-rank(value), Rank(value))`. |
giá trị | Các tensor để phân chia. |
numSplit | Số cách chia. Phải chia đều `value.shape[split_dim]`. |
Trả lại
- một phiên bản mới của Split
Danh sách công khai< Đầu ra <T>> đầu ra ()
Chúng là các tensor có hình dạng giống hệt nhau, 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à `values.shape[split_dim] / num_split`.