code: def test_forward_works_on_nd_input(self):
      highway = Highway(2, 2)
      input_tensor = torch.ones(2, 2, 2)
      output = highway(input_tensor)
      assert output.size() == (2, 2, 2)
  
file_path: magnitude/pymagnitude/third_party/allennlp/tests/modules/highway_test.py
imports: from __future__ import absolute_import
  from numpy.testing import assert_almost_equal
  import torch
  from allennlp.modules import Highway
  from allennlp.common.testing import AllenNlpTestCase
label_desc: Use image node_id 2 for calling the TestHighway obj's underlying member method code with example usage: obj.test_forward_works_on_nd_input() without return types