How to set weight randomly for TCN output

hello everyone,
I am using multi slice-TCN model in my program. So each TCN generate the output and then multiply with weight and finally the combine all TCN’s result. equation like this (output = TCN’output * weight). So my question is that “How can i randomly initial the weight for TCN”. I am sending the screen shot. I am sending also my hand draw picture because i need like this picture.Thank your very much.

Hi @Kuldeep_Rana! You can use torch.nn.init.uniform_ method to initialize your paramter. See this link