PyTorch là một thư viện máy học cho phép người dùng làm việc với các tensor. Tensor là cấu trúc dữ liệu thiết yếu có thể có nhiều loại dữ liệu và hình dạng khác nhau. Đôi khi, người dùng có thể muốn thực hiện thao tác mở rộng trên tensor để mở rộng kích thước của chúng. Hoạt động mở rộng cho phép người dùng lặp lại một tensor dọc theo các kích thước được chỉ định. PyTorch cung cấp thuộc tính “expand()” lấy tensor và danh sách các kích thước làm đầu vào. Nó trả về một tensor mới có cùng dữ liệu nhưng có kích thước khác nhau.
Bài viết này sẽ minh họa phương pháp sử dụng thao tác mở rộng trên tensor trong PyTorch.
Làm cách nào để sử dụng thao tác mở rộng trong PyTorch?
Để sử dụng thao tác mở rộng trong PyTorch, hãy xem các bước được cung cấp bên dưới:
- Nhập thư viện PyTorch
- Tạo một tensor mong muốn
- Xem kích thước tensor đầu vào
- Mở rộng tensor bằng cách sử dụng “ mở rộng() ' thuộc tính
- Hiển thị tensor mở rộng và kích thước của nó
Bước 1: Nhập thư viện PyTorch
Đầu tiên, nhập “ ngọn đuốc ” thư viện để sử dụng thao tác mở rộng:
nhập khẩu ngọn đuốc
Bước 2: Tạo một Tensor
Sau đó, tạo một tensor mong muốn bằng cách sử dụng nút “ ngọn đuốc.tensor() ” hoạt động và in các phần tử của nó. Ở đây, chúng tôi đang tạo như sau “ hàng chục tenxơ:
hàng chục = ngọn đuốc. tenxơ ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
in ( hàng chục )
Ở đầu ra bên dưới, tensor đã được tạo thành công:
Bước 3: Xem kích thước Tenor đầu vào
Tiếp theo, xem kích thước của “ Hàng chục ” tensor sử dụng “ kích cỡ() ' thuộc tính:
in ( 'Kích thước của Tenor:' , Hàng chục. kích cỡ ( ) )Theo đầu ra bên dưới, kích thước của tensor là 3 × 1:
Bước 4: Mở rộng Tenor
Bây giờ, hãy sử dụng “ mở rộng() ” để thực hiện thao tác mở rộng và mở rộng tensor sang một chiều mới. Cần phải cung cấp tensor và danh sách các kích thước làm đầu vào. Ở đây, chúng tôi đang mở rộng tensor lên kích thước 3 × 4:
Exp_tens = Hàng chục. mở rộng ( 3 , 4 )Bước 5: Hiển thị Tensor mở rộng và kích thước của nó
Cuối cùng, in các phần tử của tensor mở rộng và kích thước của nó:
in ( Exp_tens )in ( Exp_tens. kích cỡ ( ) )
Đầu ra bên dưới hiển thị tensor mở rộng và kích thước của nó, tức là 3 × 4. Điều này chỉ ra rằng hoạt động mở rộng đã được thực hiện thành công:
Chúng tôi đã giải thích phương pháp hiệu quả để sử dụng thao tác mở rộng 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 thao tác mở rộng trong PyTorch, trước tiên, hãy nhập thư viện torch. Sau đó, tạo tensor mong muốn và xem các phần tử cũng như kích thước của nó. Tiếp theo, sử dụng “ mở rộng() Thuộc tính ” để mở rộng tensor đầu vào. Cuối cùng, in tensor mở rộng và xem kích thước của nó. Bài viết này đã trình bày phương pháp sử dụng thao tác mở rộng trên tensor trong PyTorch.