Constants that should not be transferred to another device.
Useful for operations that should only happen only on the CPU
Nodes that should not be transferred to another device
Useful for operations that should only happen only on the CPU
Precision with which float tensors should be loaded. If 16 is specified, all 32 bit floats are casted to 16 bit floats.
Defaults to 32.
Generated using TypeDoc
If the model should be loaded in training or inference mode. In inference mode, no gradients can be supported and the model expects tensors as input.
In training mode, gradients can be computed and the inputs should be variables (although with noGrad: true if no gradient is needed for the respective input)