Spec-Zone.ru › PyTorch 2

UninitializedBuffer

class torch.nn.parameter.UninitializedBuffer(requires_grad=False, device=None, dtype=None) [source]

Буфер, который не инициализирован.

Неинициализированный буфер — это особый случай torch.Tensor, где форма данных всё ещё неизвестна.

В отличие от torch.Tensor, неинициализированные параметры не содержат данных, и попытка получить доступ к некоторым свойствам, таким как их форма, вызовет ошибку выполнения. Единственные операции, которые могут быть выполнены с неинициализированным параметром, — это изменение его типа данных, перемещение на другое устройство и преобразование в обычный torch.Tensor.

По умолчанию устройство или тип данных, используемые при материализации буфера, можно задать во время создания, например, device='cuda'.

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.parameter.UninitializedBuffer.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API