摘要: pytorch针对batch只有矩阵乘法torch.bmm(),没有针对batch的Hadamard product,即不存在所谓的torch.bmul()之类的函数 可以使用万能的torch.einsum()公式 a=torch.arange(24).view(2, 3, 4) b=torch.o 阅读全文
posted @ 2019-12-09 10:21 拎壶冲AR 阅读(1032) 评论(0) 推荐(0) 编辑