docs: def reshape(self, shape: tf.TensorShape) -> 'TensorFluent':
          '''Returns a TensorFluent for the reshape operation with given `shape`.
  
          Args:
              shape: The output's shape.
  
          Returns:
              A TensorFluent wrapping the reshape operation.
          '''
          t = tf.reshape(self.tensor, shape)
          scope = self.scope.as_list()
          batch = self.batch
          return TensorFluent(t, scope, batch=batch)