Learn Before
Concept
Forward Propagation Output Device Allocation
During the forward propagation of a neural network model, the resulting output tensor is automatically allocated to the exact same hardware device where the input tensor and the model parameters currently reside. Because cross-device operations are restricted, if a model and its inputs are explicitly placed on a specific GPU, the evaluation result will natively inhabit that same GPU without requiring any additional programmatic transfer commands.
0
1
Updated 2026-05-09
Tags
D2L
Dive into Deep Learning @ D2L