PyTorch е библиотека за машинно обучение, която позволява на потребителите да работят с тензори. Тензорите са основни структури от данни, които могат да имат различни типове данни и форми. Понякога потребителите може да искат да извършат операцията за разширяване на тензори, за да разширят техния размер. Операцията за разширяване позволява на потребителите да повтарят тензор по определени размери. PyTorch предоставя атрибута “expand()”, който приема тензор и списък с размери като входни данни. Той връща нов тензор, който има същите данни, но с различни измерения.
Тази статия ще илюстрира метода за използване на операцията за разширяване на тензори в PyTorch.
Как да използвам операцията Expand в PyTorch?
За да използвате операцията за разширяване в PyTorch, вижте посочените по-долу стъпки:
- Импортиране на библиотека на PyTorch
- Създайте желан тензор
- Вижте размера на входния тензор
- Разширете тензора с помощта на „ разширяване () ' атрибут
- Показване на разширен тензор и неговия размер
Стъпка 1: Импортирайте библиотеката на PyTorch
Първо импортирайте „ факла ”, за да използвате операцията за разширяване:
импортиране факла
Стъпка 2: Създайте тензор
След това създайте желания тензор, като използвате „ torch.tensor() ” и отпечатайте нейните елементи. Тук създаваме следното „ десетки ” тензор:
десетки = факла. тензор ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
печат ( десетки )
В резултата по-долу тензорът е създаден успешно:
Стъпка 3: Вижте размера на входния тензор
След това вижте размера на създадения по-горе „ десетки ' тензор с помощта на ' размер () ' атрибут:
печат ( „Размер на тензора:“ , десетки. размер ( ) )Според изхода по-долу размерът на тензора е 3×1:
Стъпка 4: Разширете Tensor
Сега използвайте „ разширяване () ” за извършване на операцията за разширяване и разширяване на тензора до ново измерение. Изисква се да предоставите тензора и списъка с размери като вход. Тук разширяваме тензора до размер 3×4:
Exp_tens = десетки. разширяване ( 3 , 4 )Стъпка 5: Показване на разширен тензор и неговия размер
И накрая, отпечатайте елементите на разширения тензор и неговия размер:
печат ( Exp_tens )печат ( Exp_tens. размер ( ) )
Изходът по-долу показва разширения тензор и неговия размер, т.е. 3×4. Това показва, че разширената операция е извършена успешно:
Обяснихме ефективния метод за използване на операцията за разширяване в PyTorch
Забележка : Можете да получите достъп до нашия бележник Google Colab тук връзка .
Заключение
За да използвате операцията за разширяване в PyTorch, първо импортирайте библиотеката на факела. След това създайте желания тензор и вижте неговите елементи и размер. След това използвайте „ разширяване () ” за разширяване на входния тензор. И накрая, отпечатайте разширения тензор и вижте размера му. Тази статия демонстрира метода за използване на операцията за разширяване на тензори в PyTorch.