Class: Tensorflow::TensorSpec

Inherits:
BatchableTypeSpec show all
Defined in:
lib/tensorflow/tensor_spec.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(shape, dtype) ⇒ TensorSpec

Returns a new instance of TensorSpec.



5
6
7
8
# File 'lib/tensorflow/tensor_spec.rb', line 5

def initialize(shape, dtype)
  @shape = shape
  @dtype = dtype
end

Instance Attribute Details

#dtypeObject (readonly)

Returns the value of attribute dtype.



3
4
5
# File 'lib/tensorflow/tensor_spec.rb', line 3

def dtype
  @dtype
end

#shapeObject (readonly)

Returns the value of attribute shape.



3
4
5
# File 'lib/tensorflow/tensor_spec.rb', line 3

def shape
  @shape
end