search

ezyang / pytorch-unattached

Tensors and Dynamic neural networks in Python with strong GPU acceleration
http://pytorch.org
Other
20 stars 8 forks source link

Some ATen methods not supported in JIT #257

Open ezyang opened 6 years ago

ezyang commented 6 years ago 
        # Generate the actuall ATen call. This gets a bit tricky because of
        # TensorList arguments, and functions that are only available as methods.
        if 'namespace' in decl['method_of']:
            if has_tensorlist:
                if sum(map(is_tensor_arg, arguments)) != 1:
                    # TODO: support this
                    continue

This includes index.

apaszke commented 6 years ago

There’s an issue for this in PyTorch repo