Python backward Exemples

Exemple #1

Fichier : test.py Projet : catrove/mmdetection-sugon

import torch
import right_pool
device = torch.device("cuda:0")
a = torch.zeros([10, 10, 10, 10], device=device)
b = torch.zeros([10, 10, 10, 10], device=device)
c = right_pool.backward(a, b)
print(c)

Exemple #2

 def backward(ctx, grad_output):
     input = ctx.saved_variables[0]
     output = right_pool.backward(input, grad_output)[0]
     return output