Làm cách nào để sử dụng phương thức “torch.argmax()” trong PyTorch?

Blog này sẽ minh họa phương pháp sử dụng phương thức “torch.argmax()” trong PyTorch.

Làm cách nào để sử dụng phương thức “torch.argmax()” trong PyTorch?

Phương thức “torch.argmax()” lấy bất kỳ tensor 1D hoặc 2D nào làm đầu vào và trả về một tensor chứa chỉ số/chỉ mục của các giá trị lớn nhất dọc theo chiều đã cho.

Cú pháp của phương thức “torch.argmax()” được đưa ra dưới đây:

Để sử dụng phương pháp này trong PyTorch, hãy xem qua các ví dụ sau để hiểu rõ hơn:

Ví dụ 1: Sử dụng phương thức “torch.argmax()” với Tensor 1D

Trong ví dụ đầu tiên, chúng ta sẽ tạo một tensor 1D và sử dụng phương thức “torch.argmax()” với nó. Hãy làm theo quy trình từng bước dưới đây:

Bước 1: Nhập thư viện PyTorch

Đầu tiên, nhập “ ngọn đuốc ” để sử dụng phương thức “torch.argmax()”:

Bước 2: Tạo Tenxơ 1D

Sau đó, tạo một tensor 1D và in các phần tử của nó. Ở đây, chúng tôi đang tạo như sau “ Hàng chục1 ” tensor từ danh sách bằng cách sử dụng “ ngọn đuốc.tensor() ' chức năng:

Điều này đã tạo ra một tenxơ 1D như bên dưới:

Bước 3: Tìm chỉ số có giá trị lớn nhất

Bây giờ, hãy sử dụng “ ngọn đuốc.argmax() ” để tìm chỉ số/chỉ số của giá trị lớn nhất trong “ Hàng chục1 tenxơ:

Bước 4: In chỉ mục giá trị tối đa

Cuối cùng, hiển thị chỉ số của giá trị lớn nhất trong tensor đầu vào:

Đầu ra bên dưới hiển thị chỉ số của giá trị tối đa trong “ Hàng chục1 ” tensor tức là 4. Nghĩa là giá trị cao nhất của tensor nằm ở chỉ số thứ 4 là “ 9 ”:

Ví dụ 2: Sử dụng phương thức “torch.argmax()” với Tensor 2D

Trong ví dụ thứ hai, chúng ta sẽ tạo một tensor 2D và sử dụng phương thức “torch.argmax()” với nó. Hãy làm theo các bước được cung cấp:

Bước 1: Nhập thư viện PyTorch

Đầu tiên, nhập “ ngọn đuốc ” để sử dụng phương thức “torch.argmax()”:

Bước 2: Tạo Tensor 2D

Sau đó, sử dụng “ ngọn đuốc.tensor() ” để tạo một tensor 2D và in các phần tử của nó. Ở đây, chúng tôi đang tạo như sau “ hàng chục2 “Tăng ten 2D:

Điều này đã tạo ra một tenxơ 2D như bên dưới:

Bước 3: Tìm chỉ số có giá trị lớn nhất

Bây giờ, hãy tìm chỉ số của giá trị lớn nhất trong “ hàng chục2 ” tensor bằng cách sử dụng “ ngọn đuốc.argmax() ' chức năng:

Bước 4: In chỉ mục giá trị tối đa

Cuối cùng, hiển thị chỉ số của giá trị lớn nhất trong tensor đầu vào:

Theo kết quả đầu ra dưới đây, chỉ số của giá trị tối đa trong “ hàng chục2 ” tensor là “3”. Nghĩa là giá trị cao nhất của tensor nằm ở chỉ số thứ 3 là “ mười lăm ”:

Bước 5: Tìm chỉ số có giá trị lớn nhất dọc theo cột

Hơn nữa, người dùng cũng có thể tìm thấy các chỉ số/chỉ số của giá trị lớn nhất dọc theo mỗi cột của tensor. Ví dụ: chúng ta có thể sử dụng “ mờ=0 ” đối số với hàm “torch.argmax()”. Nó tìm các chỉ số của giá trị lớn nhất dọc theo các cột trong phần “ hàng chục2 ” tensor rồi in các chỉ số đó:

Đầu ra bên dưới hiển thị chỉ mục của các giá trị tối đa dọc theo mỗi cột của tensor:

Bước 6: Tìm chỉ số có giá trị lớn nhất dọc theo hàng

Tương tự, người dùng cũng có thể tìm chỉ số/chỉ số của giá trị lớn nhất dọc theo mỗi hàng của tensor. Ví dụ: sử dụng “ mờ=1 ” đối số với hàm “torch.argmax()” để tìm chỉ số của giá trị lớn nhất dọc theo các hàng trong tensor “Tens2” rồi in các chỉ số đó:

Có thể xem các chỉ mục của giá trị tối đa dọc theo mỗi hàng của tenxơ “Tens2” bên dưới:

Chúng tôi đã giải thích một cách hiệu quả phương pháp sử dụng phương thức “torch.argmax()” trong PyTorch.

Ghi chú : Bạn có thể truy cập Sổ tay Google Colab của chúng tôi tại đây liên kết .

Phần kết luận

Để sử dụng phương thức “torch.argmax()” trong PyTorch, trước tiên, hãy nhập “ ngọn đuốc ' thư viện. Sau đó, tạo tensor 1D hoặc 2D mong muốn và xem các phần tử của nó. Tiếp theo, sử dụng “ ngọn đuốc.argmax() ” để tìm/tính các chỉ số/chỉ số của giá trị lớn nhất trong tensor. Ngoài ra, người dùng cũng có thể tìm chỉ số của giá trị lớn nhất dọc theo mỗi hàng hoặc cột trong tensor bằng cách sử dụng “ lờ mờ ' lý lẽ. Cuối cùng, hiển thị chỉ số của giá trị lớn nhất trong tensor đầu vào. Blog này đã minh họa phương pháp sử dụng phương thức “torch.argmax()” trong PyTorch.