code: def test_var(self):
      input = Input(shape=[2, 20])
      time = TimeDistributed(layer=Dense(30))(input)
      t1 = time.index_select(1, 0)
      t2 = time.index_select(1, 1)
      diff = auto.abs(t1 - t2)
      assert diff.get_output_shape() == (None, 30)
      assert diff.get_input_shape() == (None, 30)
      model = Model(input, diff)
      data = np.random.uniform(0, 1, [10, 2, 20])
      output = model.forward(data)
      print(output.shape)
  
file_path: analytics-zoo/pyzoo/test/zoo/pipeline/autograd/test_model.py
imports: from test.zoo.pipeline.utils.test_utils import ZooTestCase
  from zoo.pipeline.api.keras.layers import *
  from zoo.pipeline.api.keras.models import *
  import zoo.pipeline.api.autograd
label_desc: Use image node_id 1 for calling the TestAutoGradModel obj's underlying member method code with example usage: obj.test_var() without return types