Tile

Описание

Info

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

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

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

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

def __init__(self, axis, times, name=None):

Параметры

Параметр Возможные типы Описание По умолчанию
axis int Ось для заполнения. -
times int Количество повторений. -
name str Имя слоя. None

Пояснения

-

Примеры

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

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

Info

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

>>> batchsize, maps, insize = 1, 1, 5
>>> ata = gpuarray.to_gpu(np.random.randint(0, 9, (batchsize, maps, insize)).astype(np.float32))
>>> data
[[[0. 3. 0. 6. 0.]]]
>>> axis, times = 0, 3
>>> tile = Tile(axis=axis, times=times)
>>> tile(data)
[[[0. 3. 0. 6. 0.]]

 [[0. 3. 0. 6. 0.]]

 [[0. 3. 0. 6. 0.]]]