pytaco.tensor_pow

pytaco.tensor_pow(t1, t2, out_format, dtype=None)

Computes the t1**t2 element-wise.

  • If the two tensors are equal order, performs the operation element-wise

  • If the two tensors have order N and M and N > M, requires the last M dimensions of the tensor with order N be equal to the dimensions of the tensor with order M in order to broadcast.

The tensor class implements __pow__ using this method.

Parameters
t1, t2: tensors, array_like

tensors or array_like input operands.

out_format: format, mode_format

  • If a format is specified, the result tensor is stored in the format out_format.

  • If a mode_format is specified, the result the result tensor has a with all of the dimensions stored in the mode_format passed in.

dtype: Datatype, optional

The datatype of the output tensor.

Returns
power: tensor

The element wise power of the input tensors broadcasted as required.

Notes

The inner dimensions of the input tensor is broadcasted along the dimensions of whichever tensor has a higher order.