CoderOnly
CoderOnly

Reputation: 1742

tf.concat operation in PyTorch

torch.stack is not what I am searching.

I am looking for the Tensorflow's concat operation for Pytorch.

I have searched the doc http://pytorch.org/docs/0.3.0/

Upvotes: 0

Views: 2637

Answers (1)

gntoni
gntoni

Reputation: 487

I belive torch.cat is what you are looking for.

http://pytorch.org/docs/master/torch.html#torch.cat

Upvotes: 2

Related Questions