basic_ops.py 文件源码

def forward(self, a):
        if self.tensor_power:
            self.fw_result = torch.pow(self.constant, a)
            return result
        else:
            self.save_for_backward(a)
            return a.pow(self.constant)