Подходит ли Torch.nn.Linear для функций, подобных кривой?

Question

Я занимаюсь регрессионным тренингом в Pytorch.

Я использую линейный модуль в Pytorch для создания скрытого слоя -

import torch as T 
hid1 = T.nn.Linear(13, 10)  # 13-(10-10)-1
hid2 = T.nn.Linear(10, 10)
oupt = T.nn.Linear(10, 1)

Моя функция может быть прямой линией или кривой, в зависимости от набора данных.

Подходит ли T.nn.Linear и для кривых?Если нет, что использовать, если вы ожидаете кривой?