#
王胜鹏
PyTorch训练神经网络 PyTorch训练神经网络
PyTorch训练神经网络 可以使用torch.nn包来构建神经网络。 nn包则依赖于autograd包来定义模型并对它们求导。一个nn.Module包含各个层和一个forward(input)方法,该方法返回output。 如图这个神经网
2021-08-14
pytorch深度学习基础 pytorch深度学习基础
pytorch深度学习基础 Tensor对象及其运算 Tensor对象是一个维度任意的矩阵,但是一个Tensor中所有元素的数据类型必须一致。torch包含的数据类型和普遍编程语言的数据类型类似,包含浮点型,有符号整型和无符号整形,这些类型
2021-08-13