nngen icon indicating copy to clipboard operation
nngen copied to clipboard

position embedding? i cannot understand,the code?

Open henbucuoshanghai opened this issue 10 months ago • 3 comments

half = dim // 2 freqs = torch.exp( -math.log(max_period) * torch.arange(start=0, end=half, dtype=torch.float32) / half ).to(device=timesteps.device) args = timesteps[:, None].float() * freqs[None] embedding = torch.cat([torch.cos(args), torch.sin(args)], dim=-1)

henbucuoshanghai avatar Apr 19 '24 10:04 henbucuoshanghai

can u explain it,tks

henbucuoshanghai avatar Apr 19 '24 10:04 henbucuoshanghai

you can refer to transformer paper

xiaohu2015 avatar Apr 19 '24 12:04 xiaohu2015

whereis 10000?

henbucuoshanghai avatar Apr 20 '24 10:04 henbucuoshanghai