mindspore.train.rearrange_inputs
- mindspore.train.rearrange_inputs(func)[源代码]
此装饰器用于根据类的 indexes 属性对输入重新排列。
此装饰器目前用于
mindspore.train.Metric
类的 update 方法。- 参数:
func (Callable) - 要装饰的候选函数,其输入将被重新排列。
- 返回:
Callable,用于在函数之间调换输入。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> from mindspore.train import rearrange_inputs >>> class RearrangeInputsExample: ... def __init__(self): ... self._indexes = None ... ... @property ... def indexes(self): ... return getattr(self, '_indexes', None) ... ... def set_indexes(self, indexes): ... self._indexes = indexes ... return self ... ... @rearrange_inputs ... def update(self, *inputs): ... return inputs >>> >>> rearrange_inputs_example = RearrangeInputsExample().set_indexes([1, 0]) >>> outs = rearrange_inputs_example.update(5, 9) >>> print(outs) (9, 5)