Update TensorUtils.scala