Split

lớp cuối cùng công khai

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> Tách <T>
tạo (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.
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

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

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