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