import torch ImportLayer = torch.nn.Linear