Cast¶
Описание¶
Этот модуль выполняет функцию изменения типа данных тензора.
Инициализация¶
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))
```python
>>> data1.dtype
dtype('float32')
>>> data2.dtype
dtype('float16')
>>> cast = Cast(data1.dtype, data2.dtype)
>>> data1 = cast(data1).dtype
>>> data1.dtype
dtype('float16')