Bài viết này sẽ minh họa phương pháp thêm kích thước vào Tensor trong PyTorch.
Làm cách nào để thêm kích thước cho Tensor cụ thể trong PyTorch?
Người dùng có thể thêm kích thước cho bất kỳ Tensor nào, chẳng hạn như tensor 1D hoặc tensor 2D trong PyTorch. Để thêm kích thước mới cho tensor tại một vị trí cụ thể, hãy xem các ví dụ sau để hiểu rõ hơn:
- Ví dụ 1: Thêm kích thước vào Tensor 1D trong PyTorch
- Ví dụ 2: Thêm kích thước vào Tensor 2D trong PyTorch
Ví dụ 1: Thêm kích thước vào Tensor 1D trong PyTorch
Trong ví dụ này, chúng ta sẽ tạo một tensor 1D và thêm kích thước cho nó ở một vị trí cụ thể. Thực hiện theo các bước được liệt kê dưới đây để trình diễn thực tế:
Bước 1: Nhập thư viện
Đầu tiên, nhập thư viện đèn pin:
nhập khẩu ngọn đuốc
Bước 2: Tạo Tenxơ 1D
Sau đó, tạo một tenxơ một chiều. Ví dụ: chúng tôi đã tạo tensor sau và lưu trữ nó trong thư mục “ x ' Biến đổi:
x = ngọn đuốc. tenxơ ( [ 5 , 3 , số 8 , 2 ] )
Bước 3: Xem hình dạng Tensor
Tiếp theo, hiển thị hình tensor mới tạo để xem kích thước của nó:
Đầu ra bên dưới chỉ ra rằng tensor của chúng tôi là một chiều:
Bước 4: Thêm kích thước cho Tensor 1D
Bây giờ, hãy sử dụng “ torch.unsqueeze(đầu vào, mờ) ” để thêm kích thước cho tensor 1D tại vị trí cụ thể. Chẳng hạn, chúng ta đang thêm thứ nguyên cho tensor ở chỉ số 0:
Đây,
- “ hàng chục mới ” là biến bao gồm thứ nguyên được thêm vào.
- “ x ” là tensor đầu vào.
- “ mờ=0 ” được sử dụng để thêm thứ nguyên ở chỉ số 0.
Bước 5: Xác minh đầu ra
Cuối cùng, hãy đảm bảo rằng một chiều mới đã được thêm vào tensor hay chưa:
Trong kết quả đầu ra bên dưới, có thể thấy rằng chiều mới đã được thêm vào tensor 1D ở chỉ số 0:
Hơn nữa, người dùng còn có thể thêm kích thước cho các vị trí khác. Ở đây, chúng tôi đã thêm thứ nguyên ở chỉ mục đầu tiên:
Ví dụ 2: Thêm kích thước vào Tensor 2D trong PyTorch
Ở đây, chúng ta sẽ tạo/tạo một tensor 2D và thêm kích thước cho nó ở vị trí cụ thể. Hãy thử các bước được cung cấp để triển khai thực tế:
Bước 1: Nhập thư viện đèn pin
Đầu tiên, nhập thư viện đèn pin:
Bước 2: Tạo Tensor 2D
Sau đó, tạo một tensor hai chiều. Ví dụ: chúng tôi đã tạo tensor sau và lưu trữ nó trong thư mục “ x ' Biến đổi:
Bước 3: Xem hình dạng Tensor
Sau đó, hiển thị hình tensor mới tạo để xem kích thước của nó:
Theo đầu ra dưới đây, tensor này có hai chiều:
Bước 4: Thêm kích thước cho Tensor 2D
Bây giờ, thêm kích thước cho tensor 2D tại vị trí cụ thể bằng cách sử dụng “đuốc.unsqueeze(đầu vào, mờ)” chức năng. Chẳng hạn, chúng ta đang thêm thứ nguyên cho tensor ở chỉ số 0:
Bước 5: Xác minh đầu ra
Cuối cùng, xác minh xem kích thước mới đã được thêm vào tensor 2D hay chưa:
Đầu ra bên dưới cho biết kích thước mới đã được thêm thành công vào tensor 2D ở chỉ số 0:
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 .
Chúng tôi đã giải thích một cách hiệu quả phương pháp thêm thứ nguyên vào tensor trong PyTorch bằng các ví dụ khác nhau.
Phần kết luận
Để thêm kích thước cho tensor trong PyTorch, trước tiên, hãy nhập thư viện PyTorch. Sau đó, tạo một tensor 1D hoặc 2D và xem kích thước của nó. Sau đó, thêm kích thước cho tensor ở vị trí cụ thể bằng cách sử dụng nút “ torch.unsqueeze(đầu vào, mờ) ' chức năng. Người dùng cần truyền tensor đầu vào và vị trí chỉ mục mong muốn làm tham số cho hàm này. Bài viết này đã minh họa phương pháp thêm kích thước cho tensor trong PyTorch.