Split

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

แยกเทนเซอร์ออกเป็นเทนเซอร์ "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>> ตัววนซ้ำ ()

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

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