buildTensorflow icon indicating copy to clipboard operation
buildTensorflow copied to clipboard
verified publisher icon karanchahal

Adding SubtractOperation and AverageOperation

Open uditarora opened this issue 7 years ago • 0 comments

  • SubtractOperation allows subtraction between two tensors and enables gradient to flow through the subtraction
  • AverageOperation allows taking the average of a vector of tensors and enables gradient to flow through the average

The average operation will be useful in computing losses across a batch of data.

uditarora avatar Apr 11 '19 09:04 uditarora