If I have a class
class Operation():
def __init__(self):
self.layer = nn.Linear(16*16,8,bias=True)
def eval(self,input):
return = torch.relu(self.layer(input))
How can I create a 2D torch tensor of this type ? For example, the equivalent of the follwing in numpy;
matrix = np.empty((x_dim,y_dim),dtype=object)
for x in range(x_dim):
for y in range(y_dim):
matrix[x,y] = Operation()
I have tried in pytorch but seems like object is not a valid data type…
If it is really not possible is there another way it can be done ? Using ModuleList or ModuleDict while being able to refer to objects within as coordinates ?