PyTorch
|
#include <ATen/core/IListRef.h>
#include <ATen/core/Tensor.h>
#include <c10/core/DeviceGuard.h>
#include <c10/core/ScalarType.h>
Go to the source code of this file.
Namespaces | |
namespace | at |
Functions | |
c10::optional< Device > | at::device_of (const Tensor &t) |
Return the Device of a Tensor, if the Tensor is defined. More... | |
c10::optional< Device > | at::device_of (const c10::optional< Tensor > &t) |
c10::optional< Device > | at::device_of (ITensorListRef t) |
Return the Device of a TensorList, if the list is non-empty and the first Tensor is defined. More... | |