SplitV คลาสสุดท้ายสาธารณะ
แยกเทนเซอร์ออกเป็นเทนเซอร์ "num_split" ตามมิติเดียว
ค่าคงที่
สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
คงที่ <T ขยาย TType > SplitV <T> | สร้าง ( ขอบเขต ขอบเขต ค่า ตัวดำเนินการ <T> ตัวถูก ดำเนินการ <? ขยาย TNumber > ขนาดแยก ตัวดำเนินการ < TInt32 > แกน ยาว numSplit) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ SplitV ใหม่ |
ตัววนซ้ำ < ตัวดำเนินการ <T>> | ตัววนซ้ำ () |
รายการ < เอาท์พุต <T>> | เอาท์พุท () เทนเซอร์ที่มีรูปร่างตรงกับ "ค่า" ยกเว้นตามแนว "แกน" ซึ่งมีขนาด `size_splits[i]` |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
ค่าคงที่: "SplitV"
วิธีการสาธารณะ
สร้าง SplitV <T> คงที่สาธารณะ (ขอบเขต ขอบเขต , ค่า ตัวดำเนินการ <T>, ตัวดำเนินการ <? ขยาย TNumber > sizeSplits, ตัวดำเนินการ < TInt32 > แกน, ยาว numSplit)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ SplitV ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ค่า | เทนเซอร์ที่จะแยก |
ขนาดแยก | รายการที่มีขนาดของเทนเซอร์เอาท์พุตแต่ละตัวตามมิติการแยก ต้องรวมเป็นมิติของค่าตาม split_dim สามารถมีหนึ่ง -1 ที่บ่งชี้ว่ามิตินั้นจะถูกอนุมาน |
แกน | 0-D. มิติที่จะแยกออกไป ต้องอยู่ในช่วง `[-rank(value), rank(value))` |
การส่งคืน
- ตัวอย่างใหม่ของ SplitV
รายการสาธารณะ < เอาท์พุต <T>> เอาท์พุต ()
เทนเซอร์ที่มีรูปร่างตรงกับ "ค่า" ยกเว้นตามแนว "แกน" ซึ่งมีขนาด `size_splits[i]`