PyTorchలో “torch.argmax()” పద్ధతిని ఎలా ఉపయోగించాలి?

ఈ బ్లాగ్ PyTorchలో “torch.argmax()” పద్ధతిని ఉపయోగించే పద్ధతిని ఉదాహరణగా చూపుతుంది.

PyTorchలో “torch.argmax()” పద్ధతిని ఎలా ఉపయోగించాలి?

“torch.argmax()” పద్ధతి ఏదైనా 1D లేదా 2D టెన్సర్‌ని ఇన్‌పుట్‌గా తీసుకుంటుంది మరియు ఇచ్చిన పరిమాణంతో పాటు గరిష్ట విలువల సూచికలు/సూచికలను కలిగి ఉన్న టెన్సర్‌ను అందిస్తుంది.

“torch.argmax()” పద్ధతి యొక్క వాక్యనిర్మాణం క్రింద ఇవ్వబడింది:

PyTorchలో ఈ పద్ధతిని ఉపయోగించడానికి, మంచి అవగాహన కోసం క్రింది ఉదాహరణలను చూడండి:

ఉదాహరణ 1: 1D టెన్సర్‌తో “torch.argmax()” పద్ధతిని ఉపయోగించండి

మొదటి ఉదాహరణలో, మేము 1D టెన్సర్‌ని సృష్టిస్తాము మరియు దానితో “torch.argmax()” పద్ధతిని ఉపయోగిస్తాము. కింది దశల వారీ విధానాన్ని అనుసరించండి:

దశ 1: PyTorch లైబ్రరీని దిగుమతి చేయండి

మొదట, దిగుమతి చేసుకోండి ' మంట “torch.argmax()” పద్ధతిని ఉపయోగించడానికి లైబ్రరీ:

దశ 2: 1D టెన్సర్‌ని సృష్టించండి

అప్పుడు, 1D టెన్సర్‌ని సృష్టించి, దాని మూలకాలను ప్రింట్ చేయండి. ఇక్కడ, మేము ఈ క్రింది వాటిని సృష్టిస్తున్నాము ' పదాలు1 ''ని ఉపయోగించి జాబితా నుండి టెన్సర్ torch.tensor() 'ఫంక్షన్:

ఇది క్రింద చూసినట్లుగా 1D టెన్సర్‌ని సృష్టించింది:

దశ 3: గరిష్ట విలువ యొక్క సూచికలను కనుగొనండి

ఇప్పుడు, 'ని ఉపయోగించండి torch.argmax() ''లో గరిష్ట విలువ యొక్క సూచిక/సూచికలను కనుగొనడానికి ఫంక్షన్ పదాలు1 ”టెన్సర్:

దశ 4: గరిష్ట విలువ యొక్క సూచికను ముద్రించండి

చివరగా, ఇన్‌పుట్ టెన్సర్‌లో గరిష్ట విలువ సూచికను ప్రదర్శించండి:

దిగువ అవుట్‌పుట్ 'లో గరిష్ట విలువ యొక్క సూచికను చూపుతుంది పదాలు1 ” టెన్సర్ అంటే, 4. అంటే టెన్సర్ యొక్క అత్యధిక విలువ 4వ ఇండెక్స్‌లో ఉందని అర్థం 9 ”:

ఉదాహరణ 2: 2D టెన్సర్‌తో “torch.argmax()” పద్ధతిని ఉపయోగించండి

రెండవ ఉదాహరణలో, మేము 2D టెన్సర్‌ని సృష్టిస్తాము మరియు దానితో “torch.argmax()” పద్ధతిని ఉపయోగిస్తాము. అందించిన దశలను అనుసరించండి:

దశ 1: PyTorch లైబ్రరీని దిగుమతి చేయండి

మొదట, దిగుమతి చేసుకోండి ' మంట “torch.argmax()” పద్ధతిని ఉపయోగించడానికి లైబ్రరీ:

దశ 2: 2D టెన్సర్‌ని సృష్టించండి

అప్పుడు, 'ని ఉపయోగించండి torch.tensor() 2D టెన్సర్‌ని సృష్టించి, దాని ఎలిమెంట్‌లను ప్రింట్ చేయడానికి ఫంక్షన్. ఇక్కడ, మేము ఈ క్రింది వాటిని సృష్టిస్తున్నాము ' పదుల 2 “2D టెన్సర్:

ఇది క్రింద చూసినట్లుగా 2D టెన్సర్‌ని సృష్టించింది:

దశ 3: గరిష్ట విలువ యొక్క సూచికలను కనుగొనండి

ఇప్పుడు, 'లో గరిష్ట విలువ యొక్క సూచికను కనుగొనండి పదుల 2 'టెన్సర్'ని ఉపయోగించడం ద్వారా torch.argmax() 'ఫంక్షన్:

దశ 4: గరిష్ట విలువ యొక్క సూచికను ముద్రించండి

చివరగా, ఇన్‌పుట్ టెన్సర్‌లో గరిష్ట విలువ సూచికను ప్రదర్శించండి:

దిగువ అవుట్‌పుట్ ప్రకారం, 'లో గరిష్ట విలువ యొక్క సూచిక పదుల 2 ”టెన్సర్ “3”. టెన్సర్ యొక్క అత్యధిక విలువ 3వ ఇండెక్స్‌లో ఉందని దీని అర్థం “ పదిహేను ”:

దశ 5: నిలువు వరుసలతోపాటు గరిష్ట విలువ యొక్క సూచికలను కనుగొనండి

అంతేకాకుండా, వినియోగదారులు టెన్సర్ యొక్క ప్రతి నిలువు వరుసలో గరిష్ట విలువల సూచికలు/సూచికలను కూడా కనుగొనగలరు. ఉదాహరణకు, మనం ఉపయోగించవచ్చు ' మసక = 0 'torch.argmax()' ఫంక్షన్‌తో ఆర్గ్యుమెంట్. ఇది 'లోని నిలువు వరుసలతోపాటు గరిష్ట విలువల సూచికలను కనుగొంటుంది పదుల 2 ”టెన్సర్ ఆపై ఆ సూచికలను ప్రింట్ చేస్తుంది:

దిగువ అవుట్‌పుట్ టెన్సర్ యొక్క ప్రతి నిలువు వరుసలో గరిష్ట విలువల సూచికలను చూపుతుంది:

దశ 6: అడ్డు వరుసలలో గరిష్ట విలువ యొక్క సూచికలను కనుగొనండి

అదేవిధంగా, వినియోగదారులు టెన్సర్ యొక్క ప్రతి అడ్డు వరుసలో గరిష్ట విలువల సూచికలు/సూచికలను కూడా కనుగొనవచ్చు. ఉదాహరణకు, 'ని ఉపయోగించండి మసక = 1 'Tens2' టెన్సర్‌లోని వరుసల పొడవునా గరిష్ట విలువల సూచికలను కనుగొని, ఆపై ఆ సూచికలను ప్రింట్ చేయడానికి 'torch.argmax()' ఫంక్షన్‌తో ఆర్గ్యుమెంట్:

'Tens2' టెన్సర్ యొక్క ప్రతి అడ్డు వరుసలో గరిష్ట విలువ సూచికలను క్రింద చూడవచ్చు:

మేము PyTorchలో “torch.argmax()” పద్ధతిని ఉపయోగించాల్సిన పద్ధతిని సమర్థవంతంగా వివరించాము.

గమనిక : మీరు ఇందులో మా Google Colab నోట్‌బుక్‌ని యాక్సెస్ చేయవచ్చు లింక్ .

ముగింపు

PyTorchలో “torch.argmax()” పద్ధతిని ఉపయోగించడానికి, ముందుగా, “ని దిగుమతి చేయండి మంట ' గ్రంధాలయం. అప్పుడు, కావలసిన 1D లేదా 2D టెన్సర్‌ని సృష్టించి, దాని మూలకాలను వీక్షించండి. తరువాత, 'ని ఉపయోగించండి torch.argmax() ”టెన్సర్‌లోని గరిష్ట విలువల సూచికలు/సూచికలను కనుగొనడానికి/గణించడానికి పద్ధతి. అంతేకాకుండా, వినియోగదారులు 'ని ఉపయోగించి టెన్సర్‌లోని ప్రతి అడ్డు వరుస లేదా నిలువు వరుసలో గరిష్ట విలువ యొక్క సూచికలను కూడా కనుగొనవచ్చు మసకగా ” వాదన. చివరగా, ఇన్‌పుట్ టెన్సర్‌లో గరిష్ట విలువ సూచికను ప్రదర్శించండి. ఈ బ్లాగ్ PyTorchలో “torch.argmax()” పద్ధతిని ఉపయోగించే పద్ధతిని ఉదాహరణగా చూపింది.