开发者
下载
[object Object][object Object][object Object]
  • Ascend 950PR/Ascend 950DT:支持
[object Object][object Object]
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
[object Object][object Object]
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
[object Object][object Object]

将一个torch.Tensor转换为pypto.Tensor。可显式指定转换后的pypto.Tensor的名称。可将转换后的pypto.Tensor的指定维度标记为动态维度,用于表示该维度在后续编译/运行阶段可变。

[object Object]
[object Object]
[object Object][object Object]undefined
[object Object]

返回转换后的pypto.Tensor。

[object Object]
  • 入参tensor类型必须为torch.Tensor或其子类。
  • 入参tensor在指定内存格式的顺序下是连续的(tensor.is_contiguous() == True)。
  • 入参tensor支持如下数据类型(dtype):
    • torch.float16
    • torch.bfloat16
    • torch.float32
    • torch.float64
    • torch.int8
    • torch.uint8
    • torch.int16
    • torch.uint16
    • torch.int32
    • torch.uint32
    • torch.int64
    • torch.uint64
    • torch.bool
[object Object]
[object Object]

结果示例如下:

[object Object]