ฉันอยู่ในปัจจุบันมีสองคนคบเพลิง Tensors, p
แล้ว x
ซึ่งทั้งคู่ต่างมีรูปร่างของ (batch_size, input_size)
.
ฉันอยากจะลองคำนว Bernoulli ปูมบันทึก likelihoods สำหรับที่ให้ข้อมูล,แล้วกลับเป็น tensor ของขนาด (batch_size)
นี่เป็นตัวอย่างของสิ่งที่ฉันอยากจะทำอย่าง: ผมมีสูตรสำหรับปูมบันทึก likelihoods ของ Bernoulli สุ่มตัวแปร:
\sum_i^d x_{i} ln(p_i) + (1-x_i) ln (1-p_i)
บอกฉัน p
Tensor:
[[0.6 0.4 0], [0.33 0.34 0.33]]
และบอกว่าฉันมี x
tensor สำหรับที่ฐานสอง inputs จากพวกนั้น probabilities:
[[1 1 0], [0 1 1]]
และฉันต้องการคำนวณปูมบันทึกความเป็นไปได้สำหรับทุกตัวอย่างซึ่งจะผลอยู่ใน:
[[ln(0.6)+ln(0.4)], [ln(0.67)+ln(0.34)+ln(0.33)]]
มันจะเป็นไปได้ที่จะทำอย่างนี้ computation โดยไม่มีการใช้สำหรับบางอย่าง?
ฉันรู้ว่าฉันสามารถใช้ torch.sum(axis=1)
ต้องทำสุดท้าย summation ระหว่างปูมบันทึกแต่มันเป็นไปได้ที่จะทำ Bernoulli ปูมบันทึก-ความเป็นไปได้ computation โดยไม่มีการใช้สำหรับบางอย่าง? หรือใช้ตอนที่ 1 สำหรับเรื่อง? ฉันกำลังพยายาม vectorize ปฏิบัติการนี้มากเท่าที่เป็นไปได้ ฉันสาบานได้\n ว่าเราสามารถใช้ลาเท็กซ์สำหรับสมกาก่อนหน้าทำบางอย่างเปลี่ยนแปลงหรือนี่เป็นอีกเว็บไซต์ของ?