import tensorflow as tf


  with tf.name_scope(name):
    tt_cores = num_dims * [None]
    for i in range(num_dims):
      curr_core_shape = (1, shape[i], 1)
      tt_cores[i] = tf.ones(curr_core_shape, dtype=dtype)

    return TensorTrain(tt_cores, shape, tt_rank)


