PReLU constructor

PReLU(
  1. int numChannels, {
  2. Device device = Device.CPU,
})

Implementation

PReLU(this.numChannels, {Device device = Device.CPU})
  : weight = Tensor.fill(
      [numChannels],
      0.25,
      requiresGrad: true,
      device: device,
    );