You'll need to make MPSGraphTensor conform to Differentiable and AdditiveArithmetic.
MPSGraphTensor
Differentiable
AdditiveArithmetic