PyTorch使用细节
model.eval() :让BatchNorm、Dropout等失效; with torch.no_grad() : 不再缓存activation,节省显存; 这是矩阵乘法: y1 tensor tensor.T
y2 tensor.matmul(tensor.T)y3 torch.rand_like(y1)
torch.matm…
鸿蒙
2024/7/20 1:35:10