Add

Описание

Info

Родительский класс: Module

Производные классы: -

Этот модуль реализует операцию поэлементного сложения входных тензоров из переданного списка.

Инициализация

def __init__(self, name=None):

Параметры

Параметр Возможные типы Описание По умолчанию
name str Имя слоя None

Пояснения

-

Примеры

Необходимые импорты.

import numpy as np
from PuzzleLib.Backend import gpuarray
from PuzzleLib.Modules import Add

Info

gpuarray необходим для правильного размещения тензора на GPU

Сформируем несколько тензоров данных, которые затем поместим в список для дальнейшего поэлементного суммирования.

d, h, w = 2, 3, 2
data1 = gpuarray.to_gpu(np.arange(d * h * w).reshape((d, h, w)).astype(np.float32))
data2 = data1.copy()
data3 = data1.copy()
print(data1)
[[[ 0.  1.]
  [ 2.  3.]
  [ 4.  5.]]

 [[ 6.  7.]
  [ 8.  9.]
  [10. 11.]]]
add = Add()
print(add([data1, data2, data3]))
[[[ 0.  3.]
  [ 6.  9.]
  [12. 15.]]

 [[18. 21.]
  [24. 27.]
  [30. 33.]]]