torch.Tensor.set_
-
Tensor.set_(source=None, storage_offset=0, size=None, stride=None) → Tensor -
Устанавливает базовое хранилище, размер и шаги. Если
sourceявляется тензором, то тензорselfбудет совместно использовать то же хранилище и иметь тот же размер и шаги, что иsource. Изменения элементов в одном тензоре будут отражаться в другом.Если
sourceявляетсяStorage, метод устанавливает базовое хранилище, смещение, размер и шаг.- Параметры:
-
- source (Тензор или Хранилище) – тензор или хранилище для использования
- storage_offset (int, необязательно) – смещение в хранилище
- size (torch.Size, необязательно) – желаемый размер. По умолчанию равен размеру источника.
- stride (кортеж, необязательно) – желаемый шаг. По умолчанию равен шагам C-contiguous.
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.Tensor.set_.html