Python max_pool3d_grad_grad 예제들

예제 #1

파일: nn_grad.py 프로젝트: adit-chandra/tensorflow

def _MaxPool3DGradGrad(op, grad):
  return (array_ops.zeros(
      shape=array_ops.shape(op.inputs[0]), dtype=op.inputs[0].dtype),
          array_ops.zeros(
              shape=array_ops.shape(op.inputs[1]), dtype=op.inputs[1].dtype),
          gen_nn_ops.max_pool3d_grad_grad(
              op.inputs[0],
              op.inputs[1],
              grad,
              op.get_attr("ksize"),
              op.get_attr("strides"),
              padding=op.get_attr("padding"),
              data_format=op.get_attr("data_format").decode()))

예제 #2

def _MaxPool3DGradGrad(op, grad):
    return (array_ops.zeros(shape=array_ops.shape(op.inputs[0]),
                            dtype=op.inputs[0].dtype),
            array_ops.zeros(shape=array_ops.shape(op.inputs[1]),
                            dtype=op.inputs[1].dtype),
            gen_nn_ops.max_pool3d_grad_grad(
                op.inputs[0],
                op.inputs[1],
                grad,
                op.get_attr("ksize"),
                op.get_attr("strides"),
                padding=op.get_attr("padding"),
                data_format=op.get_attr("data_format").decode()))