torch.nn - Neural Networks, PyTorch Documentation Authors, 2024 (PyTorch Foundation) - Official documentation for PyTorch's neural network module, detailing layer definitions, expected input/output shapes, and parameters for various layers such as Conv2d and Linear.
torch.Tensor, PyTorch Documentation Authors, 2024 - Official documentation covering PyTorch tensors, their properties, creation, manipulation methods (like view, unsqueeze), and the rules for tensor operations including broadcasting.
Deep Learning with PyTorch, Eli Stevens, Luca Antiga, and Thomas Viehmann, 2020 (Manning Publications) - A practical guide to building deep learning models using PyTorch, offering insights into model architecture implementation and effective tensor handling practices.
Convolutional Neural Networks for Visual Recognition (CS231n) Lecture Notes, Stanford University CS231n Course Staff, 2023 (Stanford University) - Lecture notes providing an in-depth explanation of convolutional neural network architectures, including clear formulas and examples for calculating output dimensions after various layers.