Archive for the ‘pyro’ Category

Advanced index in pytorch and pyro

Wednesday, January 15th, 2020

Sometimes the broadcasting of the multidimensional index with multidimensional tensors in PyTorch is confusing. To clarify this, I try to summarize some typical cases with different strategies.

Posted in pyro, pytorch | 1 Comment »