Вот что я понимаю:
- представляет данные, обычно многомерный массив, и поддерживает такие операции, как сложение, умножение и вычисление градиента.
Код: Выделить всё
torch.Tensor - — базовый класс для всех моделей нейронных сетей в PyTorch. Он может содержать слои, параметры и определяет метод front().
Код: Выделить всё
torch.nn.Module
- Могу ли я использовать torch.Tensor для непосредственного определения моделей?
- Почему мы всегда создаем подкласс torch.nn.Module для нейронных сетей, а не просто используем Тензоры?
- Как Модуль и Тензор взаимодействуют во время обучения?
Подробнее здесь: https://stackoverflow.com/questions/798 ... in-pytorch