Pytorch 课程

• pytorch-deep-learning
• Youtube
• learnpytorch
• pytorch
• 可视化tensorflow

tips

• 在 PyTorch 中，你可以通过外部方括号 ([) 的数量来判断张量的维数，而且你只需要计算其中一侧的数。
• if else
  • device = "cuda" if torch.cuda.is_available() else "cpu"