channel_affine

dragon.vm.torch.channel_affine(
  input,
  weight,
  bias=None,
  dim=- 1,
  end_dim=None,
  out=None
)[source]

Apply affine transformation to each channel of input.

Parameters:
Returns:

dragon.vm.torch.Tensor – The output tensor.