def METHOD_NAME():
    """Test TensorVM encoder"""
    num_components = 24
    resolution = 32
    in_dim = 3
    out_dim = 3 * num_components
    encoder = encodings.TensorVMEncoding(num_components=num_components, resolution=resolution)
    assert encoder.get_out_dim() == out_dim
    in_tensor = torch.ones((3, in_dim))
    encoded = encoder(in_tensor)
    assert encoded.shape == (3, out_dim)
    in_tensor = torch.ones((6, 3, in_dim))
    encoded = encoder(in_tensor)
    assert encoded.shape == (6, 3, out_dim)
    encoder.upsample_grid(resolution=64)
    encoder.upsample_grid(resolution=128)