向量相同"/>
Torch:WX+b中,b的维度应与WX的列向量相同
self.B=Parameter(torch.FloatTensor(h_dim)) # (16, )# (16, 8285) * (8285, 8285) -> (16, 8285) -> (8285, 16)
output=(output.transpose(1,0)*temp_drop).transpose(1,0)
output+=self.B # (8285, 16) + (16,)
更多推荐
Torch:WX+b中,b的维度应与WX的列向量相同
发布评论