摘要:
1. 设置打印精 Pytorch中tensor打印的数据长度需要使用torch.set_printoptions(precision=xx)进行设置,否则打印的长度会很短,给人一种精度不够的错觉: >>> import torch >>> a=torch.tensor([1/3]) >>> a te 阅读全文
posted @ 2022-04-13 21:27
帆吖
阅读(1479)
评论(0)
推荐(0)
摘要:
1. 二维矩阵乘法 , 其中 , , 输出 的维度是。该函数一般只用来计算两个二维矩阵的矩阵乘法,而且不支持broadcast操作。 2. 三维带Batch矩阵乘法 由于神经网络训练一般采用mini-batch,经常输入的是三维带batch矩阵,所以提供 ,其中 , , 输出 的维度是 。该函数的两 阅读全文
posted @ 2022-04-13 16:22
帆吖
阅读(2417)
评论(0)
推荐(0)
浙公网安备 33010602011771号