You are viewing a read-only archive of the Blogs.Harvard network. Learn more.

~ Archive for pyro ~

Advanced index in pytorch and pyro


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.

Log in