昇腾社区首页
中文
注册

(beta)torch_npu.npu_slice

函数原型

[object Object]

功能说明

从张量中提取切片。

注意:[object Object] 该接口不支持反向计算。

参数说明

  • self (Tensor) - 输入张量。
  • offsets (ListInt) - 数据类型:int32,int64。
  • size (ListInt) - 数据类型:int32,int64。

支持的型号

  • [object Object]Atlas 训练系列产品[object Object]
  • [object Object]Atlas A2 训练系列产品[object Object]
  • [object Object]Atlas A3 训练系列产品[object Object]
  • [object Object]Atlas 推理系列产品[object Object]

调用示例

[object Object]