Augmented Tensor Decomposition with Stochastic Optimization