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()
>>> add([data1, data2, data3])
[[[ 0.  3.]
  [ 6.  9.]
  [12. 15.]]

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