View a markdown version of this page

TensorData - SageMaker Training Session

TensorData

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

Contents

data

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

Type: JSON value

Required: Yes

dtype

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

Type: String

Required: Yes

shape

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

Type: Array of integers

Required: No

See Also

For more information about using this API in one of the language-specific AWS SDKs, see the following: