Овај блог ће бити пример методе за коришћење методе „торцх.аргмак()“ у ПиТорцх-у.
Како користити метод „торцх.аргмак()“ у ПиТорцх-у?
Метода “торцх.аргмак()” узима било који 1Д или 2Д тензор као улаз и враћа тензор који садржи индексе/индексе максималних вредности дуж дате димензије.
Синтакса методе „торцх.аргмак()“ је дата у наставку:
бакља. аргмак ( < инпут_тенсор > )
Да бисте користили овај метод у ПиТорцх-у, прођите кроз следеће примере ради бољег разумевања:
Пример 1: Користите метод “торцх.аргмак()” са 1Д тензором
У првом примеру ћемо креирати 1Д тензор и користити методу „торцх.аргмак()“ са њим. Хајде да пратимо доњу процедуру корак по корак:
Корак 1: Увезите ПиТорцх библиотеку
Прво увезите „ бакља ” да користите метод „торцх.аргмак()”:
увоз бакљаКорак 2: Креирајте 1Д тензор
Затим креирајте 1Д тензор и одштампајте његове елементе. Овде креирамо следеће „ Тенс1 ” тензор са листе користећи „ торцх.тенсор() ” функција:
Тенс1 = бакља. тензор ( [ 5 , 0 , - 8 , 1 , 9 , 7 ] )
принт ( Тенс1 )
Ово је створило 1Д тензор као што се види у наставку:
Корак 3: Пронађите индексе максималне вредности
Сада користите „ торцх.аргмак() ” функција за проналажење индекса/индекса максималне вредности у „ Тенс1 ” тензор:
Т1_инд = бакља. аргмак ( Тенс1 )Корак 4: Одштампајте индекс максималне вредности
На крају, прикажите индекс максималне вредности у улазном тензору:
принт ( 'Индекси:' , Т1_инд )Доњи излаз показује индекс максималне вредности у „ Тенс1 ” тензор тј. 4. То значи да је највећа вредност тензора на 4. индексу који је „ 9 ”:
Пример 2: Користите метод “торцх.аргмак()” са 2Д тензором
У другом примеру, креираћемо 2Д тензор и са њим ћемо користити методу „торцх.аргмак()“. Пратимо наведене кораке:
Корак 1: Увезите ПиТорцх библиотеку
Прво увезите „ бакља ” да користите метод „торцх.аргмак()”:
увоз бакљаКорак 2: Креирајте 2Д тензор
Затим користите „ торцх.тенсор() ” функција за креирање 2Д тензора и штампање његових елемената. Овде креирамо следеће „ Тенс2 „2Д тензор:
Тенс2 = бакља. тензор ( [ [ 4 , 1 , - 7 ] , [ петнаест , 6 , 0 ] , [ - 7 , 9 , 2 ] ] )принт ( Тенс2 )
Ово је створило 2Д тензор као што се види у наставку:
Корак 3: Пронађите индексе максималне вредности
Сада пронађите индекс максималне вредности у „ Тенс2 “ тензор коришћењем “ торцх.аргмак() ” функција:
Т2_инд = бакља. аргмак ( Тенс2 )Корак 4: Одштампајте индекс максималне вредности
На крају, прикажите индекс максималне вредности у улазном тензору:
принт ( 'Индекси:' , Т2_инд )Према доњем излазу, индекс максималне вредности у „ Тенс2 ” тензор је „3”. То значи да је највећа вредност тензора на 3. индексу који је „ петнаест ”:
Корак 5: Пронађите индексе максималне вредности дуж колона
Штавише, корисници такође могу пронаћи индексе/индексе максималних вредности дуж сваке колоне тензора. На пример, можемо користити „ дим=0 ” аргумент са функцијом „торцх.аргмак()”. Проналази индексе максималних вредности дуж колона у „ Тенс2 ” тензор и затим штампа те индексе:
цол_индек = бакља. аргмак ( Тенс2 , замутити = 0 )принт ( „Индекси у колонама:“ , цол_индек )
Доњи излаз показује индексе максималних вредности дуж сваке колоне тензора:
Корак 6: Пронађите индексе максималне вредности дуж редова
Слично, корисници такође могу пронаћи индексе/индексе максималних вредности дуж сваког реда тензора. На пример, користите „ дим=1 ” аргумент са функцијом „торцх.аргмак()“ да бисте пронашли индексе максималних вредности дуж редова у тензору „Тенс2“, а затим одштампали те индексе:
ров_индек = бакља. аргмак ( Тенс2 , замутити = 1 )принт ( „Индекси у редовима:“ , ров_индек )
Индекси максималне вредности дуж сваког реда тензора „Тенс2“ се могу видети у наставку:
Ефикасно смо објаснили метод за коришћење методе „торцх.аргмак()“ у ПиТорцх-у.
Белешка : Овде можете приступити нашој Гоогле Цолаб бележници линк .
Закључак
Да бисте користили метод „торцх.аргмак()“ у ПиТорцх-у, прво увезите „ бакља ” библиотека. Затим креирајте жељени 1Д или 2Д тензор и погледајте његове елементе. Затим користите „ торцх.аргмак() ” метод за проналажење/израчунавање индекса/индекса максималних вредности у тензору. Штавише, корисници такође могу пронаћи индексе максималне вредности дуж сваког реда или колоне у тензору користећи „ замутити ' расправа. На крају, прикажите индекс максималне вредности у улазном тензору. Овај блог је илустровао метод за коришћење методе „торцх.аргмак()“ у ПиТорцх-у.