แยกเทนเซอร์อินพุตตัวแปรทรัพยากรในทุกมิติ
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]]
คลาสที่ซ้อนกัน
ระดับ | ReadVariableXlaSplitND.Options | คุณลักษณะที่จำเป็นสำหรับการ ReadVariableXlaSplitND |
วิธีการสาธารณะ
คง <T> ReadVariableXlaSplitND <T> | สร้าง ( ขอบเขต ขอบเขต Operand <?> ทรัพยากรชั้น <T> T, ยาว N, รายการ <ยาว> numSplits, ตัวเลือก ... ตัวเลือก) เมธอดของโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ ReadVariableXlaSplitND ใหม่ |
Iterator < Operand <T >> | iterator () |
รายการ < เอาท์พุท <T >> | เอาท์พุท () |
คง ReadVariableXlaSplitND.Options | paddings (รายการ <ยาว> paddings) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง ReadVariableXlaSplitND <T> สร้าง ( ขอบเขต ขอบเขต Operand <?> ทรัพยากรชั้น <T> T, ยาว N, รายการ <ยาว> numSplits, ตัวเลือก ... ตัวเลือก)
เมธอดของโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ ReadVariableXlaSplitND ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ทรัพยากร | ตัวแปรทรัพยากรของเทนเซอร์อินพุตเพื่อแบ่งตามมิติข้อมูลทั้งหมด } out_arg { ชื่อ: "เอาต์พุต" คำอธิบาย: < |
numSplits | จำนวนวิธีแยกต่อมิติ ขนาดรูปร่างต้องแบ่งเท่าๆ กัน |
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ ReadVariableXlaSplitND
สาธารณะคง ReadVariableXlaSplitND.Options paddings (รายการ <ยาว> paddings)
พารามิเตอร์
paddings | รายการเสริมของช่องว่างด้านขวาต่อมิติของเทนเซอร์อินพุตที่จะใช้ก่อนแยก สามารถใช้เพื่อทำให้มิติสามารถหารได้เท่าๆ กัน |
---|
แยกเทนเซอร์อินพุตตัวแปรทรัพยากรในทุกมิติ
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]]
คลาสที่ซ้อนกัน
ระดับ | ReadVariableXlaSplitND.Options | คุณลักษณะที่จำเป็นสำหรับการ ReadVariableXlaSplitND |
วิธีการสาธารณะ
คง <T> ReadVariableXlaSplitND <T> | สร้าง ( ขอบเขต ขอบเขต Operand <?> ทรัพยากรชั้น <T> T, ยาว N, รายการ <ยาว> numSplits, ตัวเลือก ... ตัวเลือก) เมธอดของโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ ReadVariableXlaSplitND ใหม่ |
Iterator < Operand <T >> | iterator () |
รายการ < เอาท์พุท <T >> | เอาท์พุท () |
คง ReadVariableXlaSplitND.Options | paddings (รายการ <ยาว> paddings) |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง ReadVariableXlaSplitND <T> สร้าง ( ขอบเขต ขอบเขต Operand <?> ทรัพยากรชั้น <T> T, ยาว N, รายการ <ยาว> numSplits, ตัวเลือก ... ตัวเลือก)
เมธอดของโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ ReadVariableXlaSplitND ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ทรัพยากร | ตัวแปรทรัพยากรของเทนเซอร์อินพุตเพื่อแบ่งตามมิติข้อมูลทั้งหมด } out_arg { ชื่อ: "เอาต์พุต" คำอธิบาย: < |
numSplits | จำนวนวิธีแยกต่อมิติ ขนาดรูปร่างต้องแบ่งเท่าๆ กัน |
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ ReadVariableXlaSplitND
สาธารณะคง ReadVariableXlaSplitND.Options paddings (รายการ <ยาว> paddings)
พารามิเตอร์
paddings | รายการเสริมของช่องว่างด้านขวาต่อมิติของเทนเซอร์อินพุตที่จะใช้ก่อนแยก สามารถใช้เพื่อทำให้มิติสามารถหารได้เท่าๆ กัน |
---|