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