This is useful while building a model in PyTorch as you have to specify the input and output shape for each layer, which might be an issue for complex networks.
確定! 回上一頁