为您找到相关结果38个
pytorch cnn 识别手写的字实现自建图片数据_python_脚本之家
class CNN(nn.Module): def __init__(self): super(CNN, self).__init__() self.conv1 = nn.Sequential( # input shape (1, 28, 28) nn.Conv2d( in_channels=1, # input height out_channels=16, # n_filters kernel_size=5, # filter size stride=1, # filter movement/step padding=2,...
www.jb51.net/article/1404...htm 2024-5-22
pytorch 实现情感分类问题小结_python_脚本之家
#length_to_mask作用是根据批次中每个序列的长度生成mask矩阵,以便处理长度不一致的序列,忽略掉比较短的序列的无效部分 #src_key_padding_mask的参数正好与length_to_mask的结果相反(无自注意力的部分为true) attention_mask = length_to_mask(lengths.to('cpu')) == False #根据批次中的每个序列长度生成mask矩...
www.jb51.net/article/2751...htm 2024-5-22