Sure,
import numpy as np
import torch.nn as nn
arr = np.arange(1,10)
layer = nn.Linear(arr[-1],arr[5])
Error, expected (int,…) didn’t match because some of the arguments have invalid types :(numpy.int64,numpy.int64) torch.FloatTensor viewed_tensor …