Implement get_dtype
and get_device_type
for tensors
#248
Labels
autograd
Tasks towards the online training / automatic differentiation feature
enhancement
New feature or request
These, in combination with the existing
get_rank
,get_shape
, andget_device_index
, will allow us to create one tensor from another using the same setup.This will be useful for autograd because the default is to compute the gradient w.r.t. a tensor of ones of the appropriate shape.
We should also make
get_device_index
a method oftorch_tensor
, as withget_rank
andget_shape
.The text was updated successfully, but these errors were encountered: