Свързани публикации 'pytorch'
Разбиране на индексирането с pytorch gather
Опитвали ли сте някога да използвате функцията за събиране на pytorch? Направих го и беше много трудно. Самата функция е доста полезна, но разбирането как да я използвате може да бъде болезнено.
И така, каква е целта на функцията за събиране? Docs казва:
torch.gather ( input , dim , index , out=None , sparse_grad=False ) → Tensor
Gathers values along an axis specified by dim.
Така че събира стойности по оста. Но как се различава от обикновеното индексиране? Когато използвате..