lớp cuối cùng công khai SplitV
Tách một tensor thành các tensor `num_split` dọc theo một chiều.
Hằng số
Sợi dây | OP_NAME | Tên của op này, được biết đến bởi công cụ lõi TensorFlow |
Phương pháp công khai
tĩnh <T mở rộng TType > 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
Hằng số
Chuỗi cuối cùng tĩnh công khai OP_NAME
Tên của op này, được biết đến bởi công cụ lõi TensorFlow
Giá trị không đổi: "SplitV"
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 <? mở rộng TNumber > sizeSplits, Toán hạng < TInt32 > trục, 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]`.