That’s a great explanation, thanks!
What should I do in case my masking tensor is actually a very large tensor with a binary pattern that I load in the argument section of the init function?
Can I somehow project a tensor I load when initiating the neural net onto forward pass?