Cast

Описание

Info

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

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

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

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

def __init__(self, intype, outtype, name=None):

Параметры

Параметр	Возможные типы	Описание	По умолчанию
intype	dtype	Входной тип данных	-
outtype	dtype	Выходной тип данных	None
name	str	Имя слоя	None

Пояснения

-

Примеры

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

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

Info

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

batchsize, size, maps = 2, 3, 2
data1 = gpuarray.to_gpu(np.arange(batchsize * size * maps).reshape((batchsize, size, maps)).astype(np.float32))
data2 = gpuarray.to_gpu(np.arange(batchsize * size * maps).reshape((batchsize, size, maps)).astype(np.float16))

print(data1.dtype)

dtype('float32')

print(data2.dtype)

dtype('float16')

cast = Cast(data1.dtype, data2.dtype)
data1 = cast(data1)
print(data1.dtype)

dtype('float16')