linear¶
dragon.vm.torch.nn.functional.linear(
input,
weight,
bias=None
)[source]¶Apply the linear transformation to input.
\[\text{out} = \text{input} \times \text{weight}^{T} + \text{bias} \]- Parameters:
- input (dragon.vm.torch.Tensor) – The input tensor.
- weight (dragon.vm.torch.Tensor) – The weight tensor.
- bias (dragon.vm.torch.Tensor, optional) – The bias tensor.
- Returns:
dragon.vm.torch.Tensor – The output tensor.
See also