scripts/models.py
def forward(self, input):
        """
        Input of shape (samplesxtime, dims)
        """
        embed = torch.cat((torch.cos(input[:, :self.angle_dims]), 
                           torch.sin(input[:, :self.angle_dims]), 
                           input[:, self.angle_dims:]), dim=-1)
        return self.net(embed)

        Input of shape (samplesxtime, dims)
        
Input of shape (samplesxtime, dims)
Input of shape ( samplesxtime dims )
def forward(self, input):
        embed = torch.cat((torch.cos(input[:, :self.angle_dims]), 
                           torch.sin(input[:, :self.angle_dims]), 
                           input[:, self.angle_dims:]), dim=-1)
        return self.net(embed)
def forward ( self , input ) : embed = torch . cat ( ( torch . cos ( input [ : , : self . angle_dims ] ) , torch . sin ( input [ : , : self . angle_dims ] ) , input [ : , self . angle_dims : ] ) , dim = - 1 ) return self . net ( embed )
Input of shape (samplesxtime, dims)
Input of shape ( samplesxtime dims )
"""
        Input of shape (samplesxtime, dims)
        """
self
input