code: def test_pow(self, inputs, gc, dc):
      op = core.CreateOperator(
          "ATen", ["S"], ["Z"], operator="pow", exponent=2.0
      )
  
      def ref(X):
          return [np.square(X)]
  
      self.assertReferenceChecks(gc, op, inputs, ref)
  
file_path: pytorch/caffe2/contrib/aten/aten_test.py
imports: from caffe2.python import core
  from hypothesis import given
  import caffe2.python.hypothesis_test_util
  import hypothesis.strategies
  import numpy
label_desc: Use image node_id 3 for calling the TestATen obj's underlying member method code with example usage: obj.test_pow(inputs, gc, dc) and returns: list