векторам, потом результат аггрегируется и выдается ответ. Как распараллелить этот процесс?
То есть, например, если есть arr=np.zeros((100,50)
Раннее был определен self.lin = nn.Linear(50,50)
Нужно в форварде сделать как-нибудь так, чтобы вот не было так:
arr_res = []
for vect in arr:
____arr_res.append(self.lin(vect))
Собирай вектора в большой батч, пропуская через линейный слой и сплит после этого делай
Обсуждают сегодня