channel_affine

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

Apply affine transformation along the channels.

Parameters:
Returns:

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