torch.Tensor.set_

Tensor.set_(source=None, storage_offset=0, size=None, stride=None) → Tensor

Задаёт нижележащее хранилище, размер и шаги. Если source — это тензор, тензор self будет использовать то же хранилище и иметь тот же размер и те же шаги, что и source. Изменения элементов одного тензора будут отражаться на другом.

Если source — это Storage, метод задаёт нижележащее хранилище, смещение, размер и шаг.

Параметры:

• source (Tensor или Storage) – тензор или хранилище, которое нужно использовать
• storage_offset (int, необязательный) – смещение в хранилище
• size (torch.Size, необязательный) – требуемый размер. По умолчанию используется размер исходного объекта.
• stride (tuple, необязательный) – требуемый шаг. По умолчанию используются шаги для непрерывного в памяти расположения в формате C.