channel_affine

dragon.channel_affine(
  inputs,
  axis=- 1,
  end_axis=None,
  **kwargs
)[source]

Apply affine transformation to each channel of input.

Parameters:
  • inputs (Sequence[dragon.Tensor]) – The input, weight and optional bias tensor.
  • axis (int, optional, default=-1) – The first channel axis.
  • end_axis (int, optional) – The last channel axis.
Returns:

dragon.Tensor – The output tensor.