Разница между torch.nn.Module и torch.Tensor в PyTorch?Python

Программы на Python
Anonymous
Разница между torch.nn.Module и torch.Tensor в PyTorch?

Сообщение Anonymous »

Я изучаю PyTorch и вижу два распространенных класса: torch.Tensor и torch.nn.Module. Я немного не понимаю их различий и того, когда их использовать.
Вот что я понимаю:
  • Код: Выделить всё

    torch.Tensor
    представляет данные, обычно многомерный массив, и поддерживает такие операции, как сложение, умножение и вычисление градиента.
  • Код: Выделить всё

    torch.nn.Module
    — базовый класс для всех моделей нейронных сетей в PyTorch. Он может содержать слои, параметры и определяет метод front().
Я не уверен:
  • Могу ли я использовать torch.Tensor для непосредственного определения моделей?
  • Почему мы всегда создаем подкласс torch.nn.Module для нейронных сетей, а не просто используем Тензоры?
  • Как Модуль и Тензор взаимодействуют во время обучения?
Может ли кто-нибудь объяснить разницу ясно на примерах?

Подробнее здесь: https://stackoverflow.com/questions/798 ... in-pytorch

Вернуться в «Python»