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.
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.