UninitializedBuffer

class torch.nn.parameter.UninitializedBuffer(requires_grad=False, device=None, dtype=None, persistent=True) [исходный код]

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

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

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

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