def _matmul_closure_factory(self, tensor):
def closure(rhs_tensor):
return torch.matmul(tensor, rhs_tensor)
return closure
微信
公众号
扫码关注公众号