torch.float_power

torch.float_power(input, exponent, *, out=None) → Tensor

Возводит input в степень exponent поэлементно, используя двойную точность. Если ни один из входных аргументов не является комплексным числом, возвращает тензор torch.float64, а если один или несколько входных аргументов являются комплексными числами, возвращает тензор torch.complex128.

Примечание

Эта функция всегда выполняет вычисления с двойной точностью, в отличие от torch.pow(), реализующей более типичное повышение типа. Это полезно, когда вычисления необходимо выполнять с более широким или точным типом данных либо когда результаты вычислений могут содержать дробные значения, не представимые во входных типах данных, например, когда целочисленное основание возводится в отрицательную целочисленную степень.

Параметры:

• input (Tensor или Number) – значение основания (или значения оснований)
• exponent (Tensor или Number) – значение показателя степени (или значения показателей степени)

Именованные аргументы:

out (Tensor, необязательно) – выходной тензор.

Пример:

>>> a = torch.randint(10, (4,))
>>> a
tensor([6, 4, 7, 1])
>>> torch.float_power(a, 2)
tensor([36., 16., 49.,  1.], dtype=torch.float64)

>>> a = torch.arange(1, 5)
>>> a
tensor([ 1,  2,  3,  4])
>>> exp = torch.tensor([2, -3, 4, -5])
>>> exp
tensor([ 2, -3,  4, -5])
>>> torch.float_power(a, exp)
tensor([1.0000e+00, 1.2500e-01, 8.1000e+01, 9.7656e-04], dtype=torch.float64)