torch.nn.functional
时间: 2023-04-29 08:03:59 浏览: 92
pytorch 中pad函数toch.nn.functional.pad()的用法
torch.nn.functional 是 PyTorch 中用于定义神经网络的前向计算过程的一个模块。它包含了一些常用的神经网络层(如卷积层、池化层等)和激活函数(如 ReLU、Sigmoid 等)。使用这些函数可以简化神经网络的定义过程。
阅读全文
pytorch 中pad函数toch.nn.functional.pad()的用法