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.