PyTorch to biblioteka do uczenia maszynowego, która umożliwia użytkownikom pracę z tensorami. Tensory to podstawowe struktury danych, które mogą mieć różne typy danych i kształty. Czasami użytkownicy mogą chcieć wykonać operację rozwijania na tensorach, aby zwiększyć ich rozmiar. Operacja rozwijania umożliwia użytkownikom powtarzanie tensora wzdłuż określonych wymiarów. PyTorch udostępnia atrybut „expand()”, który jako dane wejściowe przyjmuje tensor i listę rozmiarów. Zwraca nowy tensor, który ma te same dane, ale o innych wymiarach.
W tym artykule przedstawiono metodę użycia operacji rozwijania na tensorach w PyTorch.
Jak korzystać z operacji rozwijania w PyTorch?
Aby skorzystać z operacji rozwijania w PyTorch, wykonaj poniższe kroki:
- Importuj bibliotekę PyTorch
- Utwórz żądany tensor
- Wyświetl rozmiar tensora wejściowego
- Rozwiń tensor za pomocą „ zwiększać() ' atrybut
- Wyświetl rozwinięty tensor i jego rozmiar
Krok 1: Zaimportuj bibliotekę PyTorch
Najpierw zaimportuj „ latarka ”, aby skorzystać z operacji rozwijania:
import latarka
Krok 2: Utwórz tensor
Następnie utwórz żądany tensor za pomocą „ tensor palnika() ” i wypisz jej elementy. Tutaj tworzymy następujące „ Kilkadziesiąt tensor:
Kilkadziesiąt = latarka. napinacz ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
wydrukować ( Kilkadziesiąt )
W poniższym wyniku tensor został pomyślnie utworzony:
Krok 3: Wyświetl rozmiar tensora wejściowego
Następnie wyświetl rozmiar utworzonego powyżej „ Kilkadziesiąt ” tensor za pomocą „ rozmiar() ' atrybut:
wydrukować ( „Rozmiar tensora:” , Kilkadziesiąt. rozmiar ( ) )Zgodnie z poniższym wyjściem rozmiar tensora wynosi 3 × 1:
Krok 4: Rozwiń Tensor
Teraz użyj opcji „ zwiększać() ”, aby wykonać operację rozwijania i rozszerzyć tensor do nowego wymiaru. Jako dane wejściowe wymagane jest podanie tensora i listy rozmiarów. Tutaj rozszerzamy tensor do rozmiaru 3×4:
Exp_tens = Kilkadziesiąt. zwiększać ( 3 , 4 )Krok 5: Wyświetl rozszerzony tensor i jego rozmiar
Na koniec wydrukuj elementy rozwiniętego tensora i jego rozmiar:
wydrukować ( Exp_tens )wydrukować ( Exp_tens. rozmiar ( ) )
Poniższe dane wyjściowe pokazują rozwinięty tensor i jego rozmiar, tj. 3 × 4. Oznacza to, że rozszerzona operacja została pomyślnie wykonana:
Wyjaśniliśmy skuteczną metodę wykorzystania operacji rozwijania w PyTorch
Notatka : tutaj możesz uzyskać dostęp do naszego Notatnika Google Colab połączyć .
Wniosek
Aby skorzystać z operacji rozwijania w PyTorch, najpierw zaimportuj bibliotekę latarki. Następnie utwórz żądany tensor i wyświetl jego elementy oraz rozmiar. Następnie użyj opcji „ zwiększać() ”, aby rozwinąć tensor wejściowy. Na koniec wydrukuj rozwinięty tensor i sprawdź jego rozmiar. W tym artykule zademonstrowano metodę użycia operacji rozwijania na tensorach w PyTorch.