Crate caffe2op_sqrt

source ·

Structs

  • GetSqrtGradient
  • GetSquareRootDivideGradient
  • SqrtFunctor
    | Performs element-wise square-root | ($\sqrt{x}$) of input tensor $X$. | | Github Link: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/sqrt_op.cc |
  • SquareRootDivideOp
    | Given DATA tensor with first dimension | N and SCALE vector of the same size N produces | an output tensor with same dimensions | as DATA. | | Which consists of DATA slices. | | i-th slice is divided by sqrt(SCALE[i]) | elementwise. If SCALE[i] == 0 output | slice is identical to the input one (no | scaling) |