摘要: nn.Conv2d() 是 PyTorch 中用于定义二维卷积层(Convolutional layer)的函数,它属于 torch.nn 模块,该模块包含了构建神经网络所需的所有构建块。二维卷积层是卷积神经网络(CNN)中最基本也是最重要的组件之一,广泛用于图像和视频处理、自然语言处理等领域。 n 阅读全文
posted @ 2024-07-20 10:09 leolzi 阅读(4) 评论(0) 推荐(0) 编辑
摘要: 在PyTorch中,loss.grad_fn属性是用来访问与loss张量相关联的梯度函数的。 这个属性主要出现在使用自动微分(automatic differentiation)时,特别是在构建和训练神经网络的过程中。 当你构建一个计算图(computational graph)时,PyTorch会 阅读全文
posted @ 2024-07-20 09:40 leolzi 阅读(1) 评论(0) 推荐(0) 编辑