Неверный синтаксис с продуктом в инициализации класса? - PullRequest
0 голосов
/ 25 октября 2018

У меня есть этот класс:

NUM_IMGS, HEIGHT, WIDTH = images.shape
NUM_CLASSES = np.unique(labels).size

D_H = 100
NUM_OPT_STEPS = 100

class TwoLayerNet(torch.nn.Module):
    def __init__(self, HEIGHT*WIDTH, D_H, NUM_CLASSES):

, где высота, ширина, D_H и num_classes определены несколькими строками выше.Когда я запускаю скрипт, я получаю следующую ошибку:

  File "<ipython-input-43-cf9af12418a7>", line 2
    def __init__(self, HEIGHT*WIDTH, D_H, NUM_CLASSES):
                             ^
  SyntaxError: invalid syntax

Возможно, это глупый вопрос, но как этот синтаксис недопустим?Нужно ли определять продукт в init или передавать продукт как одну переменную?

...