每天一个tenser flow小知识

努力不脱发选手

于 2022-04-10 05:52:12 发布

阅读量397

点赞数

文章标签： tensorflow

本文链接：https://blog.csdn.net/weixin_41147129/article/details/124071941

版权

本文介绍了PyTorch中张量的view方法，用于改变维度但不复制数据，以及张量的乘法操作，包括torch.bmm()和torch.matmul()。torch.bmm()适用于3D张量，强制要求维度匹配，而torch.matmul()允许不同维度间的广播机制进行矩阵乘法。这两个函数在特定条件下等效。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 tensor .view()
改变维度但是共享内存 b改变 a的值也会改变


import torch as t
a=t.arange(0,6).view(2,3)
print(a)
 
b=a.view(-1,2)  #当某一维是-1时，会自动计算它的大小
print(b)
 
#输出
tensor([[0, 1, 2],
        [3, 4, 5]])
tensor([[0, 1],
        [2, 3],