mlx.core.power

目錄

mlx.core.power#

power(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) array#

Element-wise power operation.

Raise the elements of a to the powers in elements of b with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

參數:
  • a (array) -- Input array or scalar.

  • b (array) -- Input array or scalar.

回傳:

Bases of a raised to powers in b.

回傳型別:

array