Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

จะคำนวณฮิสโตแกรมของเทนเซอร์ใน PyTorch ได้อย่างไร?


ฮิสโตแกรมของเทนเซอร์คำนวณโดยใช้ torch.histc() . มันส่งกลับฮิสโตแกรมที่แสดงเป็นเทนเซอร์ ใช้พารามิเตอร์สี่ตัว:อินพุต, ถังขยะ, นาที และ สูงสุด . จัดเรียงองค์ประกอบเป็นช่องที่มีความกว้างเท่ากันระหว่าง นาที และ สูงสุด . โดยไม่สนใจองค์ประกอบที่เล็กกว่า นาที และมากกว่า สูงสุด .

ขั้นตอน

  • นำเข้าไลบรารีที่จำเป็น ในตัวอย่าง Python ทั้งหมดต่อไปนี้ ไลบรารี Python ที่จำเป็นคือ ไฟฉาย และ Matplotlib . ตรวจสอบให้แน่ใจว่าคุณได้ติดตั้งแล้ว

  • สร้างเทนเซอร์แล้วพิมพ์ออกมา

  • คำนวณ torch.histc(input, bins=100, min=0, max=100) . มันส่งกลับเทนเซอร์ของค่าฮิสโตแกรม ตั้งค่าถังขยะ ค่าต่ำสุด และค่าสูงสุดตามความต้องการของคุณ

  • พิมพ์ฮิสโตแกรมที่คำนวณได้ข้างต้น

  • เห็นภาพฮิสโตแกรมเป็นแผนภาพแท่ง

ตัวอย่างที่ 1

# โปรแกรม Python เพื่อคำนวณฮิสโตแกรมของเทนเซอร์# นำเข้าไลบรารีที่จำเป็น นำเข้า Tormimport matplotlib.pyplot เป็น plt# สร้าง tensorT =torch.Tensor([2,3,1,2,3,4,3,2,3, 4,3,4])print("Original Tensor T:\n",T)# คำนวณฮิสโตแกรมของ tensorhist ที่สร้างขึ้นด้านบน =torch.histc(T, bins =5, min =0, max =4)print( "ฮิสโตแกรมของ T:\n", hist)

ผลลัพธ์

Original Tensor T:tensor([2., 3., 1., 2., 3., 4., 3., 2., 3., 4., 3., 4.])ฮิสโตแกรมของ T :เทนเซอร์([0, 1., 3., 5., 3.])

ตัวอย่างที่ 2

# โปรแกรม Python เพื่อคำนวณฮิสโตแกรมของเทนเซอร์# นำเข้าไลบรารีที่จำเป็น นำเข้า Tormimport matplotlib.pyplot เป็น plt# สร้าง tensorT =torch.Tensor([2,3,1,2,3,4,3,2,3, 4,3,4])print("Original Tensor T:\n",T)# คำนวณฮิสโตแกรมของ tensorhist ที่สร้างขึ้นด้านบน =torch.histc(T, bins =5, min =0, max =4)# Visualize ฮิสโตแกรมที่คำนวณข้างต้นเป็นแท่งไดอะแกรม =5x =range(bins)plt.bar(x, hist, align='center')plt.xlabel('Bins')plt.ylabel('Frequency')plt.show() 

ผลลัพธ์

Original Tensor T:tensor([2., 3., 1., 2., 3., 4., 3., 2., 3., 4., 3., 4.])

จะคำนวณฮิสโตแกรมของเทนเซอร์ใน PyTorch ได้อย่างไร?