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.

Be Sociable, Share!

1 Comment »

  1. A WordPress Commenter

    January 15, 2020 @ 3:09 am


    Hi, this is a comment.
    To get started with moderating, editing, and deleting comments, please visit the Comments screen in the dashboard.
    Commenter avatars come from Gravatar.

Leave a Comment

Log in