Method: Torch::Tensor#type

Defined in:
lib/torch/tensor.rb

#type(dtype) ⇒ Object 


149
150
151
152
153
154
155
156
157
158
 
# File 'lib/torch/tensor.rb', line 149

def type(dtype)
  if dtype.is_a?(Class)
    raise Error, "Invalid type: #{dtype}" unless TENSOR_TYPE_CLASSES.include?(dtype)
    dtype.new(self)
  else
    enum = DTYPE_TO_ENUM[dtype]
    raise Error, "Invalid type: #{dtype}" unless enum
    _type(enum)
  end
end