torch.mtia.stream

torch.mtia.stream(stream) [source]

Обертка над контекстным менеджером StreamContext, который выбирает заданный поток.

Параметры:: stream (Stream) – выбранный поток. Этот менеджер ничего не делает, если это None.
Тип возвращаемого значения:: StreamContext

Примечание

В eager-режиме stream имеет тип класса Stream, тогда как в JIT torch.mtia.stream не поддерживается.

© 2026, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://docs.pytorch.org/docs/2.14/generated/torch.mtia.stream_function.html