Skip to content

Equivalent of tf.gather , tf.scatter, tf.update in pytorch #1965

@Sushobhan04

Description

@Sushobhan04

Is there an equivalent of tensor operations like tf.gather, tf,scatter, tf.update (tensorflow tensor operations) in pytorch without returning the same size tensor?

If yes, that would be a great help for me since I only need a specific part of the tensor. Else it would be a good operations to have, since they add versatility to the operations that can be performed on a tensor.

I am currently using tensorflow for a piece of code, but the dynamic graph of pytorch can be very useful for my application as I can dynamically select parts of the tensor, but I need to be able to tf.gather and tf.scatter. Are there alternative hacks to perform similar operations in pytorch ?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions