About the code "images = torch.cat(images, dim=0)"

sthalles / SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

https://sthalles.github.io/simple-self-supervised-learning/

MIT License

2.19k stars 457 forks source link

About the code "images = torch.cat(images, dim=0)" #47

Closed meihao5631 closed 2 years ago

meihao5631 commented 2 years ago

hi~ the "images" here is a tensor, but the first parameter in cat() is a tensor sequence. I don't understand what's going on here. Could you please explain it, thank you very much.

jingyi-y commented 2 years ago

hi I have the same question. Did you figure out why?