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