Pytorch-如果使用“重塑/查看”,渐变是否正确传输

时间:2019-01-31 20:55:34

标签: pytorch

我尝试使用谷歌搜索,但找不到以下问题的明确答案。

我有一个[180x1]张量,它是一系列标准乘法和加法运算的乘积。

我希望将其重塑为[60x3],并在X,Y,Z列上应用特定的操作。我当时正在考虑进行重塑,然后进行块操作Tensor [:,1/2/3]。

我的最终优化包括将其最小化为X,Y,Z矢量地面真值,以便在开始时找到一些最佳参数。

我想知道在这种情况下,对于“重塑”和“块”操作,渐变是否可以正确传输

0 个答案:

没有答案