摘要:
1:基本运算符(重载的+-*/或者函数add(a,b),sub(a,b),mul(a,b),div(a,b))实现基本的加减乘除。 【注】可以用eq()函数查看两种不同运算之后的维度信息是否相同,all()函数可以查看对用维度的数据元素是否相同。 2:matmul/mm/@都是矩阵相乘 【注】mm仅 阅读全文
posted @ 2021-07-10 16:48
收购阿里巴巴
阅读(345)
评论(0)
推荐(0)
摘要:
1:Broadcasting维度的自动扩张 特点:能够进行维度的自动扩张,而又不复制数据。 使用例如下图: 例如上图:A 形状为[4,32,14,14]的tensor与B 形状为[32]的tensor相加,首先需要右对齐,由于右对齐32与A最右边的 的14不匹配,故需要先手动添加两个维度生成B [3 阅读全文
posted @ 2021-07-10 10:50
收购阿里巴巴
阅读(110)
评论(0)
推荐(0)

浙公网安备 33010602011771号