Rate this Page
★ ★ ★ ★ ★

TensorDictBase.pow#

TensorDictBase.pow_(other: TensorDictBase | Tensor) → Self#

Raises every leaf to the power other in place and returns self.

The in-place version of pow(), called by td **= other. It has no default: a tensordict other must have every key of the tensordict, and its other keys are ignored.

Parameters:

other (TensorDictBase, Tensor or Number) – the exponent, paired with the leaves as in pow().

Every tensordict class and every tensorclass has this method. For the other methods, see Methods by task.