Solved by pascal_notsawo in post #2 
              
              
                For your questions, just do some test to get the answer. Here is the code that supports the answers below. 
import torch
class Model(torch.nn.Module) :
    def __init__(self) :
        super().__init__()
        torch.manual_seed(0)
        n, m = 1, 1
        A = torch.rand((n, m))
        B = torc…