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

np.random.seed(123)
batchsize, maps, insize = 1, 1, 5
data = gpuarray.to_gpu(np.random.randint(0, 9, (batchsize, maps, insize)).astype(np.float32))
print(data)

[[[2. 2. 6. 1. 3.]]]

axis, times = 0, 3
tile = Tile(axis=axis, times=times)
print(tile(data))

[[[2. 2. 6. 1. 3.]]

 [[2. 2. 6. 1. 3.]]

 [[2. 2. 6. 1. 3.]]]