Open henbucuoshanghai opened 7 months ago
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)
can u explain it,tks
you can refer to transformer paper
whereis 10000?
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)