Closed rsokl closed 3 years ago
tensor.backward(grad) now supports grad that can be broadcast into tensor. The resulting tensor.grad will be cast to the correction (floating) dtype.
tensor.backward(grad)
grad
tensor
tensor.grad
tensor.backward(grad)
now supportsgrad
that can be broadcast intotensor
. The resultingtensor.grad
will be cast to the correction (floating) dtype.