我们可以使用 arange
创建一个行向量 x
如果只想知道张量中元素的总数,可以检查它的大小(size)。
幸运的是,我们可以通过-1
来调用此自动计算出维度的功能。 即我们可以用x.reshape(-1,4)
或x.reshape(3,-1)
来取代x.reshape(3,4)
。
**运算符是求幂运算
由于a
和b
分别是3×13×1和1×21×2矩阵,如果让它们相加,它们的形状不匹配。 我们将两个矩阵广播为一个更大的3×23×2矩阵,如下所示:矩阵a
将复制列, 矩阵b
将复制行,然后再按元素相加
要(将大小为1的张量转换为Python标量),我们可以调用item
函数或Python的内置函数int float。
实现广播机制需要保证有一个相同大小的维度