Step 2：预处理
已关闭
1455-李同学-算法方向-计算机视觉-就业：否
1455-李同学-算法方向-计算机视觉-就业：否
![]()
扫二维码继续学习 二维码时效为半小时
Step 1：创建自定义数据集
w = torch.rand(16, 3, 5, 5)
= (ker_num, input_channel, ker_size, ker_size)
Input_channels:
- 黑白：1
- 彩色：3
Stocastic: 随即筛选样本
val_set: for detecting overfitting
torch.nn.function
.matmul() 取后两维相乘
unsqueeze:
正：在之前插入
负：在之后插入
.index_select(0, [0, 2])
torch.tensor([2., 3.2])
torch.FloatTensor(2, 3)
Unintialized: 未初始化的tensor
增强学习一般用 DoubleTensor