Как использовать операцию расширения в PyTorch?

PyTorch — это библиотека машинного обучения, которая позволяет пользователям работать с тензорами. Тензоры — это важные структуры данных, которые могут иметь разные типы данных и формы. Иногда пользователям может потребоваться выполнить операцию расширения тензоров, чтобы увеличить их размер. Операция расширения позволяет пользователям повторять тензор по указанным размерам. PyTorch предоставляет атрибут «expand()», который принимает в качестве входных данных тензор и список размеров. Он возвращает новый тензор с теми же данными, но с другими размерностями.

В этой статье будет продемонстрирован метод использования операции расширения тензоров в PyTorch.

Как использовать операцию расширения в PyTorch?

Чтобы использовать операцию расширения в PyTorch, выполните следующие шаги:

• Импортировать библиотеку PyTorch
• Создайте желаемый тензор
• Просмотр размера входного тензора
• Разверните тензор, используя « расширять() » атрибут
• Отображение расширенного тензора и его размера

Шаг 1. Импортируйте библиотеку PyTorch.

Сначала импортируйте « факел ” для использования операции расширения:

Шаг 2. Создайте тензор

Затем создайте желаемый тензор, используя « факел.тензор() » и распечатайте ее элементы. Здесь мы создаем следующее: Десятки тензор:

В приведенном ниже выводе тензор был успешно создан:

Шаг 3. Просмотр размера входного тензора

Далее просмотрите размер созданного выше « Десятки тензор с использованием « размер() » атрибут:

Согласно приведенному ниже выводу, размер тензора равен 3×1:

Шаг 4: Разверните тензор

Теперь используйте « расширять() » для выполнения операции расширения и расширения тензора до нового измерения. В качестве входных данных необходимо предоставить тензор и список размеров. Здесь мы расширяем тензор до размера 3×4:

Шаг 5. Отображение расширенного тензора и его размера

Наконец, напечатайте элементы расширенного тензора и его размер:

В приведенном ниже выводе показан расширенный тензор и его размер, т. е. 3 × 4. Это указывает на то, что расширенная операция выполнена успешно:

Мы объяснили эффективный метод использования операции расширения в PyTorch.

Примечание : Вы можете получить доступ к нашему блокноту Google Colab по этому адресу. связь .

Заключение

Чтобы использовать операцию расширения в PyTorch, сначала импортируйте библиотеку факела. Затем создайте нужный тензор и просмотрите его элементы и размер. Далее используйте « расширять() » Атрибут для расширения входного тензора. Наконец, распечатайте расширенный тензор и просмотрите его размер. В этой статье продемонстрирован метод использования операции расширения тензоров в PyTorch.