def METHOD_NAME(mocker):
    x = torch.ones((10, 4, 4, 4))
    indexes = torch.LongTensor([0, 1, 2])
    mock_tensor_meta = mocker.stub
    traced_x = TracedTensor.from_torch_tensor(torch.ones((10, 4, 4, 4)), mock_tensor_meta)
    traced_indexes = TracedTensor.from_torch_tensor(torch.LongTensor([0, 1, 2]), mock_tensor_meta)
    SHAPE_1 = [3, 4, 4, 4]
    SHAPE_2 = [3, 4, 4]
    assert list(x[indexes].shape) == SHAPE_1
    assert list(x[traced_indexes].shape) == SHAPE_1
    assert list(traced_x[indexes].shape) == SHAPE_1
    assert list(traced_x[traced_indexes].shape) == SHAPE_1
    assert list(x[indexes, indexes].shape) == SHAPE_2
    assert list(x[traced_indexes, traced_indexes].shape) == SHAPE_2
    assert list(traced_x[indexes, indexes].shape) == SHAPE_2
    assert list(traced_x[traced_indexes, traced_indexes].shape)
    assert list(x[indexes, traced_indexes].shape) == SHAPE_2
    assert list(traced_x[indexes, traced_indexes].shape) == SHAPE_2