Flatten

Описание

Info

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

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

Этот модуль выполняет функцию выпрямления (уплощения) тензора: тензор размерности $(N, d_1, ..., d_k)$ приводится к размерности $(N, \displaystyle\prod_{i=1}^{k}d_i)$.

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

def __init__(self, name=None):

Параметры

Параметр	Возможные типы	Описание	По умолчанию
name	str	Имя слоя	None

Пояснения

-

Примеры

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

import numpy as np
from PuzzleLib.Backend import gpuarray
from PuzzleLib.Modules import Flatten

Info

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

data = gpuarray.to_gpu(np.random.randn(10, 2, 32, 32).astype(np.float32))
print(data.shape)

(10, 2, 32, 32)

flatten = Flatten()
data = flatten(data)
print(data.shape)

(10, 2048)