Свързани публикации 'pytorch'


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