affine¶
dragon.math.
affine
(
inputs,
axis=- 1,
**kwargs
)[source]¶Apply affine transformation to input.
- Parameters:
- inputs (Sequence[dragon.Tensor]) – The input, scale and bias tensor.
- axis (Union[int, Sequence[int]], optional, default=-1) – The axis to apply.
- Returns:
dragon.Tensor – The output tensor.