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.