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

Contributors are:

Who are from:

Tags

D2L

Dive into Deep Learning @ D2L