PyTorch — это библиотека машинного обучения, которая позволяет пользователям работать с тензорами. Тензоры — это важные структуры данных, которые могут иметь разные типы данных и формы. Иногда пользователям может потребоваться выполнить операцию расширения тензоров, чтобы увеличить их размер. Операция расширения позволяет пользователям повторять тензор по указанным размерам. PyTorch предоставляет атрибут «expand()», который принимает в качестве входных данных тензор и список размеров. Он возвращает новый тензор с теми же данными, но с другими размерностями.
В этой статье будет продемонстрирован метод использования операции расширения тензоров в PyTorch.
Как использовать операцию расширения в PyTorch?
Чтобы использовать операцию расширения в PyTorch, выполните следующие шаги:
- Импортировать библиотеку PyTorch
- Создайте желаемый тензор
- Просмотр размера входного тензора
- Разверните тензор, используя « расширять() » атрибут
- Отображение расширенного тензора и его размера
Шаг 1. Импортируйте библиотеку PyTorch.
Сначала импортируйте « факел ” для использования операции расширения:
Импортировать факел
Шаг 2. Создайте тензор
Затем создайте желаемый тензор, используя « факел.тензор() » и распечатайте ее элементы. Здесь мы создаем следующее: Десятки тензор:
Десятки '=' факел. тензор ( [ [ 2 ] , [ 4 ] , [ 6 ] ] )
Распечатать ( Десятки )
В приведенном ниже выводе тензор был успешно создан:
Шаг 3. Просмотр размера входного тензора
Далее просмотрите размер созданного выше « Десятки тензор с использованием « размер() » атрибут:
Распечатать ( «Размер тензора:» , Десятки. размер ( ) )Согласно приведенному ниже выводу, размер тензора равен 3×1:
Шаг 4: Разверните тензор
Теперь используйте « расширять() » для выполнения операции расширения и расширения тензора до нового измерения. В качестве входных данных необходимо предоставить тензор и список размеров. Здесь мы расширяем тензор до размера 3×4:
Exp_tens '=' Десятки. расширять ( 3 , 4 )Шаг 5. Отображение расширенного тензора и его размера
Наконец, напечатайте элементы расширенного тензора и его размер:
Распечатать ( Exp_tens )Распечатать ( Exp_tens. размер ( ) )
В приведенном ниже выводе показан расширенный тензор и его размер, т. е. 3 × 4. Это указывает на то, что расширенная операция выполнена успешно:
Мы объяснили эффективный метод использования операции расширения в PyTorch.
Примечание : Вы можете получить доступ к нашему блокноту Google Colab по этому адресу. связь .
Заключение
Чтобы использовать операцию расширения в PyTorch, сначала импортируйте библиотеку факела. Затем создайте нужный тензор и просмотрите его элементы и размер. Далее используйте « расширять() » Атрибут для расширения входного тензора. Наконец, распечатайте расширенный тензор и просмотрите его размер. В этой статье продемонстрирован метод использования операции расширения тензоров в PyTorch.