pytorch 常用函数备用

detach()

contiguous()

  • 有些tensor并不是占用一整块内存,而是由不同的数据块组成,而tensor的view()操作依赖于内存是整块的,这时只需要执行contiguous()这个函数,把tensor变成在内存中连续分布的形式
  • https://blog.csdn.net/appleml/article/details/80143212

view()

permute()

cat()

猜你喜欢

转载自blog.csdn.net/xyz2107605729/article/details/88976373