Skip to content

Tensor

Name Mandatory Description Default Type
⬅️ Input The input of the shard, if any [Int][Float]
Output ➡️ The resulting output of the shard Tensor
Shape No The shape of the tensor to create. If not provided, the shape will be inferred from the input. none [Int]Var([Int])
Type No The data type of the tensor to create. If not provided, the data type will be inferred from the input. TensorType::F32 TensorType
GPU No If true, the tensor will be created on the GPU (if available). false Bool

Creates a tensor from a sequence (or nested sequences) of variables.