摘要: PyTorch 中的 expand 函数用于扩展张量的形状,使其在某些维度上“看起来”像被复制了多次,但实际上它不会复制数据,从而节省内存和计算资源。扩展后的张量共享原始张量的内存空间,因此原始张量和扩展后的张量是同一个数据的视图。 以下是 torch.expand 函数的一些基本用法: 1. 扩展 阅读全文
posted @ 2024-07-15 09:22 Picassooo 阅读(10) 评论(0) 推荐(0) 编辑