ExpandDims

แทรกมิติ 1 ลงในรูปร่างของเทนเซอร์

เมื่อพิจารณาเมตริกซ์ "อินพุต" การดำเนินการนี้จะแทรกมิติ 1 ที่ดัชนีมิติ "แกน" ของรูปร่าง "อินพุต" ดัชนีขนาด `แกน` เริ่มต้นที่ศูนย์ หากคุณระบุจำนวนลบสำหรับ "แกน" ระบบจะนับถอยหลังจากจุดสิ้นสุด

การดำเนินการนี้มีประโยชน์หากคุณต้องการเพิ่มมิติชุดงานให้กับองค์ประกอบเดียว ตัวอย่างเช่น หากคุณมีรูปภาพเดียวที่มีรูปร่าง `[ความสูง ความกว้าง ช่อง]` คุณสามารถทำให้เป็นชุดรูปภาพ 1 รูปด้วย `expand_dims(image, 0)` ซึ่งจะสร้างรูปร่าง `[1 ความสูง , ความกว้าง, ช่อง]`.

ตัวอย่างอื่นๆ:

# 't' is a tensor of shape [2]
 shape(expand_dims(t, 0)) ==> [1, 2]
 shape(expand_dims(t, 1)) ==> [2, 1]
 shape(expand_dims(t, -1)) ==> [2, 1]
 
 # 't2' is a tensor of shape [2, 3, 5]
 shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
 shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
 shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]
 

`-1-input.dims() <= สลัว <= input.dims()`

การดำเนินการนี้เกี่ยวข้องกับ `squeeze()` ซึ่งจะลบขนาดขนาด 1 ออก