How to implement in Matlab Deep Learning PyTorch detach or TensorFlow stop_gradient?

6 次查看(过去 30 天)
Hello!
Pytorch has a facility to detach a tensor so that it will never require a gradient, i.e. (from here):
In order to enable automatic differentiation, PyTorch keeps track of all operations involving tensors for which the gradient may need to be computed (i.e., require_grad is True). The operations are recorded as a directed graph. The detach() method constructs a new view on a tensor which is declared not to need gradients, i.e., it is to be excluded from further tracking of operations, and therefore the subgraph involving this view is not recorded.
TensorFlow has a comparable facility stop_gradient.
These are useful when one needs copies of expressions that are treated as constants and whose gradient should not be calculated during learning.
How does one implement such a thing in Matlab's Deep Learning Toolbox? (Possibly in a custom training loop)
Thx,
D

采纳的回答

Damien T
Damien T 2022-11-7
You just need to call the following. It will turn a traced dlarray into a standard Matlab variable, hence the autodiff engine will treat the new ("detached") variable as a constant.
myDetachedVar = extractdata(myVar);

更多回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Image Data Workflows 的更多信息

产品


版本

R2021a

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by