Ejemplos de gradients_v2 en Python

Lenguaje de programación: Python

Namespace/Package Name: tensorflow.python.ops.gradients_impl

Método / Función: gradients_v2

Ejemplos en hotexamples.com: 2

Python gradients_v2 - 2 ejemplos encontrados. Estos son los ejemplos en Python del mundo real mejor valorados de tensorflow.python.ops.gradients_impl.gradients_v2 extraídos de proyectos de código abierto. Puedes valorar ejemplos para ayudarnos a mejorar la calidad de los ejemplos.

Ejemplo n.º 1

Mostrar archivo

Archivo: gradient_checkpointing.py Proyecto: zijunsu/Shortcut-3D-UNet-Tensorflow2.0-BraTS-2020

def _gradients(*args, **kwargs):
    return (gradients_impl.gradients_v2(*args, **kwargs)
            if tf2.enabled() else gradients_impl.gradients(*args, **kwargs))

Ejemplo n.º 2

Mostrar archivo

 def step(c):
     x = constant_op.constant([[3.], [5.]])
     y = constant_op.constant([[2.], [4.]])
     mid = all_gather_fn([x, y])
     y = mid * c
     return gradients_impl.gradients_v2(y, [x])[0]