axpby

dragon.vm.torch.axpby(
  input,
  alpha=1.0,
  beta=1.0,
  out=None
)[source]

Compute the element-wise addition from input to output.

\[\text{out} = \alpha * \text{input} + \beta * \text{out} \]
Parameters:
  • input (dragon.vm.torch.Tensor) – The input tensor.
  • alpha (float, optional, default=1.) – The value to \(\alpha\).
  • beta (float, optional, default=1.) – The value to \(\beta\).
  • out (dragon.vm.torch.Tensor, optional) – The optional output tensor.
Returns:

dragon.vm.torch.Tensor – The output tensor.