How to shape data to pass data through a linear layer and receive the same output shape as input shape?

后端未结

关注

 0  520

I am wondering how I can receive the same input shape as an output shape for a torch linear layer?

Consider the following (failing MWE):

x = torch.randn(40


                      
              相关标签: