se3-transformer-pytorch icon indicating copy to clipboard operation
se3-transformer-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon lucidrains

Linear layer

Open abdalazizrashid opened this issue 4 years ago • 1 comments

Is there a way to add an equivariant linear layer as a head? It would be great if you could put a minimal example.

abdalazizrashid avatar Jun 07 '21 22:06 abdalazizrashid

Any linear layer w/o bias is going to be equivariant. you can just use LinearSE3(fiber_in, fiber_out)

MattMcPartlon avatar Sep 16 '21 20:09 MattMcPartlon