Укажите np.ndarray и torch.tensor как два параметра dtype. ⇐ Python

[Anonymous]

Я хотел бы указать два варианта типа данных для переменной. Примерно так:

Код: Выделить всё

class MyDataClass:

MyData: np.ndarray | torch.float

def __init__(self, as_torch):
MyData = np.arange(1,10)
if as_torch:
MyData = torch.from_numpy(MyData)

... но выдает ошибку:

Код: Выделить всё

TypeError: unsupported operand type(s) for |: 'type' and 'torch.dtype'

Я думаю, что можно было бы просто не назначать тип данных MyData, но мне интересно, есть ли лучшее решение.
Как лучше всего это сделать? обойти эту ошибку или решить ее? Заранее спасибо!
Пробовал:
np.ndarray | факел.тензор

Подробнее здесь: https://stackoverflow.com/questions/785 ... or-a-dtype