Solved by ptrblck in post #2 
              
              
                Since you are passing two values to self.c, your expected shape would be [N, 2]. 
Could you explain a bit what your use case is and how self.c should be stacked to create a [500] tensor?