摘要: np.prod()函数用来计算所有元素的乘积,对于有多个维度的数组可以指定轴,如axis=1指定计算每一行的乘积。 x.view(x.size(0), -1) 首先,在pytorch中的view()函数就是用来改变tensor的形状的,例如将2行3列的tensor变为1行6列,其中-1表示会自适应的 阅读全文
posted @ 2022-03-22 09:58 今天记笔记了吗 阅读(475) 评论(0) 推荐(0)