SplitV

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

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

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

คงที่ <T, U ขยายหมายเลข> SplitV <T>
สร้าง ( ขอบเขต ขอบเขต ค่า ตัวดำเนินการ <T> ตัวถูกดำเนินการ <U> sizeSplits แกนตัว ถูกดำเนินการ <Integer> ยาว numSplit)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ SplitV ใหม่
ตัววนซ้ำ < ตัวดำเนินการ <T>>
รายการ < เอาท์พุต <T>>
เอาท์พุท ()
เทนเซอร์ที่มีรูปร่างตรงกับ "ค่า" ยกเว้นตามแนว "แกน" ซึ่งมีขนาด `size_splits[i]`

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

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

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

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
ค่า เทนเซอร์ที่จะแยก
ขนาดแยก รายการที่มีขนาดของเทนเซอร์เอาท์พุตแต่ละตัวตามมิติการแยก ต้องรวมเป็นมิติของค่าตาม split_dim สามารถมีหนึ่ง -1 ที่บ่งชี้ว่ามิตินั้นจะถูกอนุมาน
แกน 0-D. มิติที่จะแยกออกไป ต้องอยู่ในช่วง `[-rank(value), rank(value))`
การส่งคืน
  • ตัวอย่างใหม่ของ SplitV

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

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

เทนเซอร์ที่มีรูปร่างตรงกับ "ค่า" ยกเว้นตามแนว "แกน" ซึ่งมีขนาด `size_splits[i]`