Class: Aws::SageMakerTrainingSessionRuntime::Types::TensorData

Inherits:
Struct
  • Object
show all
Defined in:
gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb

Overview

A tensor with data, data type, and shape metadata.

See Also:

Constant Summary collapse

SENSITIVE = 
[]

Instance Attribute Summary collapse

Instance Attribute Details

#data ⇒ Hash, ...

The raw tensor data as JSON. The format is determined by the dtype field: integer arrays for int64, floating-point arrays for float32.

Returns:

  • (Hash, Array, String, Numeric, Boolean) 


1111
1112
1113
1114
1115
1116
1117
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 1111

class TensorData < Struct.new(
  :data,
  :dtype,
  :shape)
  SENSITIVE = []
  include Aws::Structure
end

#dtype ⇒ String

The data type of the tensor elements, such as int64 or float32.

Returns:

  • (String) 


1111
1112
1113
1114
1115
1116
1117
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 1111

class TensorData < Struct.new(
  :data,
  :dtype,
  :shape)
  SENSITIVE = []
  include Aws::Structure
end

#shape ⇒ Array<Integer>

The dimensions of the tensor, such as [batch_size, seq_length].

Returns:

  • (Array<Integer>) 


1111
1112
1113
1114
1115
1116
1117
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 1111

class TensorData < Struct.new(
  :data,
  :dtype,
  :shape)
  SENSITIVE = []
  include Aws::Structure
end