torch.Storage

torch.Storage является псевдонимом для класса хранения, который соответствует по умолчанию типу данных (torch.get_default_dtype()). Например, если тип данных по умолчанию — torch.float, то torch.Storage соответствует torch.FloatStorage.

Классы torch.<type>Storage и torch.cuda.<type>Storage, например, torch.FloatStorage, torch.IntStorage и т. д., на самом деле никогда не создаются. Вызов их конструкторов создаёт torch.TypedStorage с соответствующим torch.dtype и torch.device. Классы torch.<type>Storage имеют все те же методы класса, что и torch.TypedStorage.

torch.TypedStorage — это непрерывный одномерный массив элементов определённого torch.dtype. Ему можно задать любой torch.dtype, и внутренние данные будут интерпретироваться соответствующим образом. torch.TypedStorage содержит torch.UntypedStorage, который хранит данные как нетипизированный массив байтов.

Каждый строковый torch.Tensor содержит torch.TypedStorage, который хранит все данные, на которые ссылается torch.Tensor.

class torch.TypedStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

bfloat16() [source]

Преобразует это хранилище к типу bfloat16

bool() [source]

Преобразует это хранилище к типу bool

byte() [source]

Преобразует это хранилище к типу byte

char() [source]

Преобразует это хранилище к типу char

clone() [source]

Возвращает копию этого хранилища

complex_double() [source]

Преобразует это хранилище к типу комплексного двойного

complex_float() [source]

Преобразует это хранилище к типу комплексного с плавающей запятой

copy_(source, non_blocking=None) [source]

cpu() [source]

Возвращает копию хранилища на процессоре, если оно не находится на процессоре.

cuda(device=None, non_blocking=False, **kwargs) [source]

Возвращает копию этого объекта в памяти CUDA.

Если этот объект уже находится в памяти CUDA и на правильном устройстве, то копия не выполняется, и возвращается исходный объект.

Параметры:

• device (int) – Идентификатор GPU назначения. По умолчанию текущее устройство.
• non_blocking (bool) – Если True и источник находится в закреплённой памяти, то копия будет асинхронной по отношению к хосту. В противном случае аргумент не имеет эффекта.
• **kwargs – Для совместимости может содержать ключ async вместо аргумента non_blocking.

Возвращаемый тип:

T

data_ptr() [source]

property device

double() [source]

Преобразует это хранилище к типу double

dtype: dtype

element_size() [source]

fill_(value) [source]

float() [source]

Преобразует это хранилище к типу float

classmethod from_buffer(*args, dtype=None, device=None, **kwargs) [source]

classmethod from_file(filename, shared=False, size=0) → Storage [source]

Если shared равно True, то память совмещена между всеми процессами. Все изменения записываются в файл. Если shared равно False, то изменения в хранилище не влияют на файл.

size – количество элементов в хранилище. Если shared равно False, то файл должен содержать как минимум size * sizeof(Type) байта (Type – тип хранилища). Если shared равно True, файл будет создан при необходимости.

Параметры:

• filename (str) – имя файла для отображения
• shared (bool) – совмещать ли память
• size (int) – количество элементов в хранилище

get_device() [source]

Возвращаемый тип:

int

half() [source]

Преобразует это хранилище к типу half

int() [source]

Преобразует это хранилище к типу int

property is_cuda

is_pinned() [source]

is_shared() [source]

is_sparse = False

long() [source]

Преобразует это хранилище к типу long

nbytes() [source]

pickle_storage_type() [source]

pin_memory() [source]

Копирует хранилище в закреплённую память, если оно ещё не закреплено.

resize_(size) [source]

share_memory_() [source]

Перемещает хранилище в общую память.

Это бесполезная операция для хранилищ, уже находящихся в общей памяти, и для хранилищ CUDA, которым не нужно перемещаться для совместного использования между процессами. Хранилища в общей памяти не могут быть изменены в размерах.

Возвращает: self

short() [source]

Преобразует это хранилище к типу short

size() [source]

tolist() [source]

Возвращает список, содержащий элементы этого хранилища

type(dtype=None, non_blocking=False) [source]

Возвращает тип, если dtype не указан, в противном случае преобразует этот объект к указанному типу.

Если этот объект уже имеет правильный тип, никакой копии не выполняется, и возвращается исходный объект.

Параметры:

• dtype (type или строка) – Желаемый тип
• non_blocking (bool) – Если True, и источник находится в закреплённой памяти, а место назначения — на GPU или наоборот, копирование выполняется асинхронно относительно процессора. В противном случае аргумент не имеет эффекта.
• **kwargs – Для совместимости, может содержать ключ async вместо аргумента non_blocking. Аргумент async устарел.

Тип возвращаемого значения:

Union[T, str]

untyped() [source]

Возвращает внутреннее torch.UntypedStorage

class torch.UntypedStorage(*args, **kwargs) [source]

bfloat16()

Преобразование этого хранилища в тип bfloat16

bool()

Преобразование этого хранилища в тип bool

byte()

Преобразование этого хранилища в тип byte

char()

Преобразование этого хранилища в тип char

clone()

Возвращает копию этого хранилища

complex_double()

Преобразование этого хранилища в тип комплексного двойного

complex_float()

Преобразование этого хранилища в тип комплексного float

copy_()

cpu()

Возвращает копию этого хранилища на CPU, если оно не находится на CPU

cuda(device=None, non_blocking=False, **kwargs)

Возвращает копию этого объекта в памяти CUDA.

Если этот объект уже находится в памяти CUDA и на правильном устройстве, то копия не выполняется, и возвращается исходный объект.

Параметры:

• device (int) – Идентификатор GPU назначения. По умолчанию - текущее устройство.
• non_blocking (bool) – Если True и источник находится в закрепленной памяти, копия будет асинхронной по отношению к хосту. В противном случае аргумент не оказывает влияния.
• **kwargs – Для совместимости, может содержать ключ async вместо аргумента non_blocking.

data_ptr()

device: device

double()

Преобразование этого хранилища в тип double

element_size()

fill_()

float()

Преобразование этого хранилища в тип float

static from_buffer()

static from_file(filename, shared=False, size=0) → Storage

Если shared равно True, память совместно используется между всеми процессами. Все изменения записываются в файл. Если shared равно False, изменения в хранилище не влияют на файл.

size — количество элементов в хранилище. Если shared равно False, файл должен содержать как минимум size * sizeof(Type) байт (Type — тип хранилища). Если shared равно True, файл будет создан при необходимости.

Параметры:

• filename (str) – имя файла для отображения
• shared (bool) – использовать общую память
• size (int) – количество элементов в хранилище

get_device()

Тип возвращаемого значения:

int

half()

Преобразование этого хранилища в тип half

int()

Преобразование этого хранилища в тип int

property is_cuda

is_pinned()

is_shared()

is_sparse: bool = False

is_sparse_csr: bool = False

long()

Преобразование этого хранилища в тип long

mps()

Возвращает копию этого хранилища на CPU, если оно не находится на CPU

nbytes()

new()

pin_memory()

Копирует хранилище в закрепленную память, если оно не закреплено.

resize_()

share_memory_()

Перемещает хранилище в общую память.

Это ничто для хранилищ, уже находящихся в общей памяти, и для хранилищ CUDA, которым не нужно перемещаться для совместного использования между процессами. Хранилища в общей памяти не могут быть изменены в размерах.

Возвращает: self

short()

Преобразование этого хранилища в тип short

size()

Тип возвращаемого значения:

int

tolist()

Возвращает список, содержащий элементы этого хранилища

type(dtype=None, non_blocking=False, **kwargs)

Возвращает тип, если dtype не предоставлен, иначе преобразует этот объект в указанный тип.

Если это уже правильный тип, то копия не выполняется, и возвращается исходный объект.

Параметры:

• dtype (type или строка) – Желаемый тип
• non_blocking (bool) – Если True, и источник находится в закрепленной памяти, а назначение — на GPU или наоборот, копия выполняется асинхронно по отношению к хосту. В противном случае аргумент не оказывает влияния.
• **kwargs – Для совместимости, может содержать ключ async вместо аргумента non_blocking. Аргумент async устарел.

untyped()

class torch.DoubleStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.float64 [source]

class torch.FloatStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.float32 [source]

class torch.HalfStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.float16 [source]

class torch.LongStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.int64 [source]

class torch.IntStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.int32 [source]

class torch.ShortStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.int16 [source]

class torch.CharStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.int8 [source]

class torch.ByteStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.uint8 [source]

class torch.BoolStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.bool [source]

class torch.BFloat16Storage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.bfloat16 [source]

class torch.ComplexDoubleStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.complex128 [source]

class torch.ComplexFloatStorage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.complex64 [source]

class torch.QUInt8Storage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.quint8 [source]

class torch.QInt8Storage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.qint8 [source]

class torch.QInt32Storage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.qint32 [source]

class torch.QUInt4x2Storage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.quint4x2 [source]

class torch.QUInt2x4Storage(*args, wrap_storage=None, dtype=None, device=None) [source]

dtype: dtype = torch.quint2x4 [source]