torch.nn.functional.linear

torch.nn.functional.linear(input, weight, bias=None)

Applies a linear transformation to the incoming data: $y = xA^T + b$.

This operator supports TensorFloat32.

Shape:

• Input: $(N, *, in\_features)$ N is the batch size, * means any number of additional dimensions

• Weight: $(out\_features, in\_features)$

• Bias: $(out\_features)$

• Output: $(N, *, out\_features)$