Mul¶
Описание¶
Этот модуль выполняет функцию поэлементного умножения тензоров.
Инициализация¶
def __init__(self, name=None):
Параметры
Параметр | Возможные типы | Описание | По умолчанию |
---|---|---|---|
name | str | Имя слоя. | None |
Пояснения
-
Примеры¶
Необходимые импорты.
>>> import numpy as np
>>> from PuzzleLib.Backend import gpuarray
>>> from PuzzleLib.Modules import Mul
Info
gpuarray
необходим для правильного размещения тензора на GPU
>>> maps, h, w = 1, 2, 2
>>> data1 = gpuarray.to_gpu(np.arange(maps * h * w).reshape((maps, h, w)).astype(np.float32))
>>> data2 = gpuarray.to_gpu(np.arange(maps * h * w).reshape((maps, h, w)).astype(np.float32))
>>> print(data1)
[[[0. 1.]
[2. 3.]]]
>>> mul = Mul()
>>> mul([data1, data2])
[[[0. 1.]
[4. 9.]]]