Mul

Описание

Info

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

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

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

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

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()
print(mul([data1, data2]))
[[[0. 1.]
  [4. 9.]]]