Das habe ich bisher verstanden:
- stellt Daten dar, normalerweise ein mehrdimensionales Array, und unterstützt Operationen wie Addition, Multiplikation und Gradientenberechnung.
Code: Select all
torch.Tensor - ist eine Basisklasse für alle neuronalen Netzwerkmodelle in PyTorch. Es kann Ebenen und Parameter enthalten und definiert eine vorwärts()-Methode.
Code: Select all
torch.nn.Module
- Kann ich Torch.Tensor verwenden, um Modelle direkt zu definieren?
- Warum verwenden wir stattdessen immer eine Unterklasse von Torch.nn.Module für neuronale Netze oder nur Tensoren verwenden?
- Wie interagieren Modul und Tensor während des Trainings?
Mobile version