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 (кортеж, необязательно) – желаемый шаг. По умолчанию используется C-непрерывный шаг.