text: import torch
  from torch import nn
  
  
  class Scale(nn.Module):
      def __init__(self, init_value=1.0):
          super(Scale, self).__init__()
          self.scale = nn.Parameter(torch.FloatTensor([init_value]))
  
      def forward(self, input):
          return input * self.scale
  