https://docs.pytorch.org/tutorials/beginner/basics/intro.html

Setup

import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision.transforms import ToTensor

Dataset

• training data
• testing data

Modal

• Building modal
• Using modal

Tensor

import numpy as np
import torch

a = np.array([[1, 2], [3, 4]])   # NumPy 数组
t = torch.tensor(a)               # 转换成张量

print(type(a))  # <class 'numpy.ndarray'>
print(type(t))  # <class 'torch.Tensor'>