Advertisement

Pytorch Tutorial 学习笔记

阅读量:
复制代码
    x_train, y_train, x_valid, y_valid = map(
    torch.tensor, (x_train, y_train, x_valid, y_valid)
    )
    
    
    AI写代码csharp

这里的map作用是什么呢?把上面的代码展开来看,实际为:

复制代码
    x_train, y_train, x_valid, y_valid = [torch.tensor(x_train), torch.tensor(y_train), torch.tensor(x_valid), torch.tensor(y_valid)]
    
    
    AI写代码csharp

持续更新…

全部评论 (0)

还没有任何评论哟~