PyTorch Forums
How does nn.linear work in cpp for multi dimension input? (torch._C._nn.linear)
C++
ptrblck
June 10, 2022, 1:38am
2
This thread
points to the CUDA implementations in case that’s helpful.
show post in topic