เลือกองค์ประกอบจาก "x" หรือ "y" ขึ้นอยู่กับ "เงื่อนไข"
เทนเซอร์ `x` และ `y` ทั้งหมดจะต้องมีรูปร่างเหมือนกัน และผลลัพธ์ก็จะมีรูปร่างนั้นเช่นกัน
เทนเซอร์ "เงื่อนไข" ต้องเป็นสเกลาร์ ถ้า "x" และ "y" เป็นสเกลาร์ หาก "x" และ "y" เป็นเวกเตอร์หรืออยู่ในอันดับที่สูงกว่า "เงื่อนไข" จะต้องเป็นสเกลาร์ ซึ่งเป็นเวกเตอร์ที่มีขนาดตรงกับมิติแรกของ "x" หรือต้องมีรูปร่างเหมือนกับ "x"
เทนเซอร์ "เงื่อนไข" ทำหน้าที่เป็นมาสก์ที่เลือกโดยพิจารณาจากค่าของแต่ละองค์ประกอบ ว่าองค์ประกอบ/แถวที่เกี่ยวข้องในเอาต์พุตควรนำมาจาก "x" (หากเป็นจริง) หรือ "y" (หากเป็นเท็จ)
หาก `เงื่อนไข` เป็นเวกเตอร์และ `x` และ `y` เป็นเมทริกซ์ที่มีอันดับสูงกว่า ระบบจะเลือกแถว (มิติภายนอก) ที่จะคัดลอกจาก `x` และ `y` หาก "เงื่อนไข" มีรูปร่างเหมือนกับ "x" และ "y" ระบบจะเลือกองค์ประกอบที่จะคัดลอกจาก "x" และ "y"
ตัวอย่างเช่น:
# 'condition' tensor is [[True, False]
# [False, True]]
# 't' is [[1, 2],
# [3, 4]]
# 'e' is [[5, 6],
# [7, 8]]
select(condition, t, e) # => [[1, 6], [7, 4]]
# 'condition' tensor is [True, False]
# 't' is [[1, 2],
# [3, 4]]
# 'e' is [[5, 6],
# [7, 8]]
select(condition, t, e) ==> [[1, 2],
[7, 8]]
วิธีการสาธารณะ
เอาท์พุต <T> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
คงที่ <T> โดยที่3 <T> | สร้าง (ขอบเขต ขอบเขต ตัวดำเนิน การ <บูลีน> เงื่อนไข ตัว ดำเนินการ <T> x ตัวดำเนินการ <T> y) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ Where3 ใหม่ |
เอาท์พุต <T> | เอาท์พุท () = `เทนเซอร์` ที่มีประเภทและรูปร่างเดียวกันกับ `x` และ `y` |
วิธีการสืบทอด
วิธีการสาธารณะ
เอาท์พุท สาธารณะ <T> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สาธารณะคงที่ โดยที่ 3 <T> สร้าง (ขอบเขต ขอบเขต , เงื่อนไข ตัวดำเนินการ <บูลีน>, ตัวดำเนินการ <T> x, ตัวดำเนินการ <T> y)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ Where3 ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
x | = `เทนเซอร์` ซึ่งอาจมีรูปร่างเหมือนกับ `เงื่อนไข` หาก "เงื่อนไข" อยู่ในอันดับที่ 1 "x" อาจมีอันดับสูงกว่า แต่มิติข้อมูลแรกต้องตรงกับขนาดของ "เงื่อนไข" |
ย | = `เทนเซอร์` ที่มีชนิดและรูปร่างเดียวกันกับ `x` |
การส่งคืน
- ตัวอย่างใหม่ของ Where3