Python Programming Language

I am looking for the most efficient way to get the Jacobian of a function through Pytorch and have … Read more Pytorch Most Efficient Jacobian/hessian Calculation

I want to find the number of non-zero elements in a tensor along a particular axis. Is there any Py… Read more Find Number Of Non-zero Elements In A Tensor Along An Aixs

The following question is not a duplicate of How to apply layer-wise learning rate in Pytorch? beca… Read more Freezing Individual Weights In Pytorch

I have: inp = torch.randn(4, 1040, 161) and I have another tensor called indices with values: ten… Read more How Can I Slice A Pytorch Tensor With Another Tensor?

I have a batch of segmentation masks of shape [5,1,100,100] (batch_size x dims x ht x wd) which I h… Read more Add Extra Dimension To An Axes

I have seen a lot of specific posts to particular case-specific problems, but no fundamental motiva… Read more What Does "runtimeerror: Cuda Error: Device-side Assert Triggered" In Pytorch Mean?