I don’t fully understand the error RuntimeError: mat1 and mat2 shapes cannot be multiplied (1x1 and 3x256). Does it say the input of linear layer should be changed to (1x1)? Tried different things in a = torch.full(shape_you_need, a) and the following error persists:
TypeError: full() received an invalid combination of arguments - got (tuple, Tensor), but expected one of:
 * (tuple of ints size, Number fill_value, *, Tensor out, torch.dtype dtype, torch.layout layout, torch.device device, bool pin_memory, bool requires_grad)
 * (tuple of ints size, Number fill_value, *, tuple of names names, torch.dtype dtype, torch.layout layout, torch.device device, bool pin_memory, bool requires_grad)