小黑新征程之LR-CNN
爱喝喜茶爱吃烤冷面的小黑黑
这个作者很懒,什么都没留下…
展开
-
小黑深入底层:batchify_with_label数据处理
import torchimport torch.autograd as autograddef batchify_with_label(input_batch_list,gpu,num_layer,volatile_flag = False): """ input_list:batch_size个7*list 0.wordid:(max_len) 1.biword_id:(max_len]) 2.---- 3.gaz_id原创 2022-03-30 12:06:07 · 409 阅读 · 0 评论 -
小黑生活丰富多彩:Attention机制
1.ScaledDotProductAttentionimport torch.nn as nnimport numpy as npclass ScaledDotProductAttention(nn.Module): def __init__(self,temperature,hidden_dim,attn_dropout = 0.1): super().__init__() self.temperature = temperature原创 2022-03-26 16:00:37 · 741 阅读 · 0 评论