سلام
برای پیدا کردن ماتریس ژاکوبین مربوط به یک لایه هیدن با توجه به ورودیش به دنبال تابع مناسبی هستم و اخیرا با torch.autograd.functional.vjp برخورد کرده ام که سه ورودی دریافت می کند (func,inputs,v) که در مسئله من آرگومان اول تابعی است که ورودی لایه هیدن را تولید می کند و دومین آرگومان برابر است با ورودی مربوط به این تابع و سومی ماتریس وزن لایه هیدن است. آیا vjp انتخاب درستی است؟
پاسخ به این سوال در فروم پای تورچ