Split

สาธารณะคลาสสุดท้าย แยก

แยกเทนเซอร์ออกเป็นเทนเซอร์ "num_split" ตามมิติเดียว

วิธีการสาธารณะ

คงที่ <T> แยก <T>
สร้าง ( ขอบเขต ขอบเขต แกนตัว ดำเนินการ <Integer> ค่าตัว ถูกดำเนินการ <T> ยาว numSplit)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการแยกใหม่
ตัววนซ้ำ < ตัวดำเนินการ <T>>
รายการ < เอาท์พุต <T>>
เอาท์พุท ()
พวกมันคือเทนเซอร์ที่มีรูปทรงเหมือนกัน ซึ่งมีรูปร่างตรงกับของ `value` ยกเว้นตามแนว `axis` ซึ่งมีขนาด `values.shape[split_dim] / num_split`

วิธีการสืบทอด

วิธีการสาธารณะ

สร้าง สาธารณะ แยก <T> แบบคงที่ (ขอบเขต ขอบเขต , แกนตัว ถูกดำเนินการ <Integer>, ค่าตัว ถูกดำเนินการ <T>, Long numSplit)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการแยกใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
แกน 0-D. มิติที่จะแยกออกไป ต้องอยู่ในช่วง `[-rank(value), rank(value))`
ค่า เทนเซอร์ที่จะแยก
จำนวนแยก จำนวนวิธีที่จะแยก ต้องหาร `value.shape[split_dim]` เท่าๆ กัน
การส่งคืน
  • ตัวอย่างใหม่ของ Split

ตัววนซ้ำสาธารณะ < ตัวดำเนินการ <T>> ตัววนซ้ำ ()

รายการสาธารณะ < เอาท์พุต <T>> เอาท์พุต ()

พวกมันคือเทนเซอร์ที่มีรูปทรงเหมือนกัน ซึ่งมีรูปร่างตรงกับของ `ค่า` ยกเว้นตามแนว `แกน` ซึ่งมีขนาด `values.shape[split_dim] / num_split`