Hi, I am looking for mapping inputs by learnable scalars, this is correct?
for batch_idx, (inputs, targets) in enumerate(trainloader):
            if use_cuda:
                inputs, targets = inputs.cuda(), targets.cuda()
                
            shape = torch.Size((batch_size, 3, 32, 32))
            scalar = torch.cuda.FloatTensor(shape)
            M1 = torch.rand(shape, out=scalar)
          
            inputs_new = inputs * M1