จะเพิ่มมิติให้กับเทนเซอร์ใน PyTorch ได้อย่างไร

บทความนี้จะยกตัวอย่างวิธีการเพิ่มมิติให้กับ Tensor ใน PyTorch

จะเพิ่มมิติให้กับเทนเซอร์เฉพาะใน PyTorch ได้อย่างไร

ผู้ใช้สามารถเพิ่มมิติให้กับ Tensor ใดก็ได้ เช่น 1D tensor หรือ 2D tensor ใน PyTorch หากต้องการเพิ่มมิติใหม่ให้กับเทนเซอร์ที่ตำแหน่งเฉพาะ โปรดดูตัวอย่างต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น:

• ตัวอย่างที่ 1: เพิ่มมิติให้กับ 1D Tensor ใน PyTorch
• ตัวอย่างที่ 2: เพิ่มมิติให้กับเทนเซอร์ 2 มิติใน PyTorch

ตัวอย่างที่ 1: เพิ่มมิติให้กับ 1D Tensor ใน PyTorch

ในตัวอย่างนี้ เราจะสร้างเทนเซอร์ 1D และเพิ่มมิติที่ตำแหน่งเฉพาะ ทำตามขั้นตอนด้านล่างเพื่อการสาธิตการใช้งานจริง:

ขั้นตอนที่ 1: นำเข้าไลบรารี
ขั้นแรก นำเข้าไลบรารี torch:

ขั้นตอนที่ 2: สร้างเทนเซอร์ 1D
จากนั้น สร้างเทนเซอร์แบบมิติเดียว ตัวอย่างเช่น เราได้สร้างเทนเซอร์ต่อไปนี้และเก็บไว้ใน ' x ' ตัวแปร:

ขั้นตอนที่ 3: ดูรูปร่างเทนเซอร์
ถัดไป แสดงรูปร่างเทนเซอร์ที่สร้างขึ้นใหม่เพื่อดูขนาด:

ผลลัพธ์ด้านล่างบ่งชี้ว่าเทนเซอร์ของเรามีมิติเดียว:

ขั้นตอนที่ 4: เพิ่มมิติให้กับ 1D Tensor
ตอนนี้ใช้ ' torch.unsqueeze (อินพุต, สลัว) ” เพื่อเพิ่มมิติให้กับเทนเซอร์ 1D ที่ตำแหน่งเฉพาะ ตัวอย่างเช่น เรากำลังเพิ่มมิติให้กับเทนเซอร์ที่ดัชนี 0:

ที่นี่,

• “ ใหม่_สิบ ” คือตัวแปรที่มีมิติที่เพิ่มเข้ามา
• “ x ” คือเทนเซอร์อินพุต
• “ สลัว=0 ” ใช้เพื่อเพิ่มมิติที่ดัชนี 0

ขั้นตอนที่ 5: ตรวจสอบผลลัพธ์
สุดท้ายนี้ ตรวจสอบให้แน่ใจว่าได้เพิ่มมิติใหม่ให้กับเทนเซอร์แล้วหรือไม่:

ในเอาต์พุตด้านล่าง สังเกตได้ว่ามีการเพิ่มมิติใหม่ให้กับเทนเซอร์ 1D ที่ดัชนี 0:

นอกจากนี้ผู้ใช้ยังสามารถเพิ่มมิติให้กับตำแหน่งอื่นได้อีกด้วย ที่นี่ เราได้เพิ่มมิติที่ดัชนีแรก:

ตัวอย่างที่ 2: เพิ่มมิติให้กับเทนเซอร์ 2 มิติใน PyTorch

ที่นี่ เราจะสร้าง/สร้างเทนเซอร์ 2D และเพิ่มมิติที่ตำแหน่งเฉพาะ ลองใช้ขั้นตอนที่ให้มาเพื่อการใช้งานจริง:

ขั้นตอนที่ 1: นำเข้าไลบรารีคบเพลิง
ขั้นแรก นำเข้าไลบรารี torch:

ขั้นตอนที่ 2: สร้างเทนเซอร์ 2D
จากนั้นจึงสร้างเทนเซอร์สองมิติ ตัวอย่างเช่น เราได้สร้างเทนเซอร์ต่อไปนี้และเก็บไว้ใน ' x ' ตัวแปร:

ขั้นตอนที่ 3: ดูรูปร่างเทนเซอร์
หลังจากนั้น ให้แสดงรูปร่างเทนเซอร์ที่สร้างขึ้นใหม่เพื่อดูขนาด:

ตามผลลัพธ์ด้านล่าง เทนเซอร์นี้เป็นสองมิติ:

ขั้นตอนที่ 4: เพิ่มมิติให้กับเทนเซอร์ 2D
ตอนนี้ ให้เพิ่มมิติให้กับเทนเซอร์ 2D ที่ตำแหน่งเฉพาะโดยใช้ “torch.unsqueeze (อินพุต, สลัว)” การทำงาน. ตัวอย่างเช่น เรากำลังเพิ่มมิติให้กับเทนเซอร์ที่ดัชนี 0:

ขั้นตอนที่ 5: ตรวจสอบผลลัพธ์
สุดท้าย ตรวจสอบว่ามีการเพิ่มมิติใหม่ให้กับเทนเซอร์ 2D หรือไม่:

ผลลัพธ์ด้านล่างบ่งชี้ว่ามิติใหม่ได้รับการเพิ่มลงในเทนเซอร์ 2D ที่ดัชนี 0 เรียบร้อยแล้ว:

บันทึก : คุณสามารถเข้าถึง Google Colab Notebook ของเราได้ที่นี้ ลิงค์ .

เราได้อธิบายวิธีการเพิ่มมิติให้กับเทนเซอร์ใน PyTorch อย่างมีประสิทธิภาพโดยใช้ตัวอย่างที่แตกต่างกัน

บทสรุป

หากต้องการเพิ่มมิติให้กับเทนเซอร์ใน PyTorch ก่อนอื่นให้นำเข้าไลบรารี PyTorch จากนั้น สร้างเทนเซอร์ 1D หรือ 2D และดูขนาดของเซ็นเซอร์ หลังจากนั้นให้เพิ่มมิติให้กับเทนเซอร์ที่ตำแหน่งเฉพาะโดยใช้ปุ่ม “ torch.unsqueeze (อินพุต, สลัว) ' การทำงาน. ผู้ใช้จำเป็นต้องส่งเทนเซอร์อินพุตและตำแหน่งดัชนีที่ต้องการเป็นพารามิเตอร์ให้กับฟังก์ชันนี้ บทความนี้ได้ยกตัวอย่างวิธีการเพิ่มมิติให้กับเทนเซอร์ใน PyTorch