torch基本功能介绍

torch的基本功能 torch:张量的相关运算,eg:创建、索引、切片、连续、转置、加减乘除等相关运算。 torch.nn:包含搭建网络层的模块(modules)和一系列的loss函数。eg.全连接、卷积、池化、BN分批处理、dropout、CrossEntropyLoss、MSLoss等。 torch.autograd:提供Tensor所有操作的自动求导方法。 torch.nn.functio
相关文章
相关标签/搜索