SplitV

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.

Phương pháp công khai

tĩnh <T, U mở rộng Số> 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.
Trình vòng lặp< Toán hạng <T>>
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

public Iterator< Toán hạng <T>> iterator ()

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]`.