Add

Description

Info

Parent class: Module

Derived classes: -

This module implements the operation of element-wise addition of input tensors from the passed list.

Initializing

def __init__(self, name=None):

Parameters

Parameter Allowed types Description Default
name str Layer name None

Explanations

-

Examples

Necessary imports.

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

Info

gpuarray is required to properly place the tensor in the GPU.

Let us form several data tensors, which will then be put in a list and summed element-wise.

>>> 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.]]]