คำเตือน: API นี้เลิกใช้งานแล้วและจะถูกลบออกใน TensorFlow เวอร์ชันในอนาคตหลังจาก การแทนที่ เสถียรแล้ว

XlaSplitND

ประชาชน XlaSplitND ชั้นสุดท้าย

แยกเมตริกซ์อินพุตในทุกมิติ

op ที่แบ่งเมตริกซ์อินพุตตามแอตทริบิวต์ num_splits ที่กำหนด แบ่งแผ่นตัวเลือก และส่งคืนสไลซ์ ชิ้นจะถูกส่งคืนในลำดับหลักแถว

op นี้อาจสร้างขึ้นผ่านบริดจ์ TPU

ยกตัวอย่างเช่นด้วย `เมตริกซ์ input`:

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
 
` num_splits`:
[2, 2]
 
และ `paddings`:
[1, 1]
 
ที่คาดว่า` outputs` คือ
[[0, 1],
  [3, 4]]
 [[2, 0],
  [5, 0]]
 [[6, 7],
  [0, 0]]
 [[8, 0],
  [0, 0]]
 

คลาสที่ซ้อนกัน

ระดับ XlaSplitND.ตัวเลือก คุณลักษณะที่จำเป็นสำหรับการ XlaSplitND

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

คง <T> XlaSplitND <T>
สร้าง ( ขอบเขต ขอบเขต Operand <T> อินพุตยาว N, รายการ <ยาว> numSplits, ตัวเลือก ... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ XlaSplitND ใหม่
Iterator < Operand <T >>
รายการ < เอาท์พุท <T >>
คง XlaSplitND.Options
paddings (รายการ <ยาว> paddings)

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

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

สาธารณะคง XlaSplitND <T> สร้าง ( ขอบเขต ขอบเขต Operand <T> อินพุตยาว N, รายการ <ยาว> numSplits, ตัวเลือก ... ตัวเลือก)

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

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
ป้อนข้อมูล อินพุตเทนเซอร์เพื่อแบ่งตามมิติข้อมูลทั้งหมด } out_arg { ชื่อ: "เอาต์พุต" คำอธิบาย: <
numSplits จำนวนวิธีแยกต่อมิติ ขนาดรูปร่างต้องแบ่งเท่าๆ กัน
ตัวเลือก ดำเนินการค่าแอตทริบิวต์ทางเลือก
คืนสินค้า
  • อินสแตนซ์ใหม่ของ XlaSplitND

ประชาชน Iterator < Operand <T >> iterator ()

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

สาธารณะคง XlaSplitND.Options paddings (รายการ <ยาว> paddings)

พารามิเตอร์
paddings รายการเสริมของช่องว่างด้านขวาต่อมิติของเทนเซอร์อินพุตที่จะใช้ก่อนแยก สามารถใช้เพื่อทำให้มิติสามารถหารได้เท่าๆ กัน