Hi, I wrote
Self.weight = nn.parameter (torch.randn ((1, channels, 1, 1)))
but, it does not work.
(TypeError: randn(): argument ‘size’ must be tuple of ints, but found element of type Tensor at pos 1)
I think all values are set as int, so how can I deal with this.
Best regards