torch:reshape
(torch:reshape a shape)
Differentiable reshape (row-major, linalg:reshape's rules: sizes must agree, one extent may be -1 and is inferred); the backward pass reshapes the gradient back to the input's shape.
(torch:reshape a shape)
Differentiable reshape (row-major, linalg:reshape's rules: sizes must agree, one extent may be -1 and is inferred); the backward pass reshapes the gradient back to the input's shape.