class_docstr: A highly optimized equivalent of labda x: Entmax15([x, 0])
class_docstr_tok: ["A","highly","optimized","equivalent","of","labda","x",":","Entmax15","(","[","x",",","0","]",")"]
code: def _backward(output, grad_output):
      gppr0, gppr1 = output.sqrt(), (1 - output).sqrt()
      grad_input = grad_output * gppr0
      q = grad_input / (gppr0 + gppr1)
      grad_input -= q * gppr0
      return grad_input
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from torch import Tensor
  from torch.autograd import Function
  from torch.jit import script
label_desc: Use image node_id 4 for calling the Entmoid15 obj's underlying member method code with example usage: obj._backward(output, grad_output) and returns: grad_input