Method: Torch::Tensor#size

Defined in:
lib/torch/tensor.rb

#size(dim = nil) ⇒ Object


92
93
94
95
96
97
98
# File 'lib/torch/tensor.rb', line 92

def size(dim = nil)
  if dim
    _size(dim)
  else
    shape
  end
end