Home / Class/ CustomLinear Class — pytorch Architecture

CustomLinear Class — pytorch Architecture

Architecture documentation for the CustomLinear class in dist_util.py from the pytorch codebase.

Class python DynamoBenchmarks

Entity Profile

Relationship Graph

Source Code

benchmarks/dynamo/dist_util.py lines 37–43

class CustomLinear(torch.nn.Module):
    def __init__(self, a, b):
        super().__init__()
        self.weight = nn.Parameter(torch.randn(a, b))

    def forward(self, x):
        return torch.mm(x, self.weight)

Domain

Source

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free