转Tensor (ToTensorStreamOp)

Java 类名：com.alibaba.alink.operator.stream.dataproc.ToTensorStreamOp
Python 类名：ToTensorStreamOp

功能介绍

将指定列转为 Alink 的张量类型。
如果指定列为 String 类型，并且值为 Alink 张量或者向量 toString 的结果，那么张量类型和形状将自动获取。
否则的话，需要指定张量类型和张量形状。

参数说明

| 名称 | 中文名称 | 描述 | 类型 | 是否必须？ | 取值范围 | 默认值 | | —- | —- | —- | —- | —- | —- | —- |

| selectedCol | 选中的列名 | 计算列对应的列名 | String | ✓ | | |

| handleInvalidMethod | 处理无效值的方法 | 处理无效值的方法，可取 error, skip | String | | “ERROR”, “SKIP” | “ERROR” |

| outputCol | 输出结果列 | 输出结果列列名，可选，默认null | String | | | null |

| reservedCols | 算法保留列名 | 算法保留列 | String[] | | | null |

| tensorDataType | 要转换的张量数据类型 | 要转换的张量数据类型。 | String | | “FLOAT”, “DOUBLE”, “INT”, “LONG”, “BOOLEAN”, “BYTE”, “UBYTE”, “STRING” | |

| tensorShape | 张量形状 | 张量的形状，数组类型。 | Long[] | | | null |

| numThreads | 组件多线程线程个数 | 组件多线程线程个数 | Integer | | | 1 |

代码示例

Python 代码

from pyalink.alink import *
import pandas as pd
useLocalEnv(1)
df = pd.DataFrame(["FLOAT#6#0.0 0.1 1.0 1.1 2.0 2.1 "])
source = StreamOperator.fromDataframe(df, schemaStr='vec string')
source.link(
    ToTensorStreamOp()
        .setSelectedCol("vec")
        .setTensorShape([2, 3])
        .setTensorDataType("float")
).print()
StreamOperator.execute()

Java 代码

import org.apache.flink.types.Row;
import com.alibaba.alink.operator.stream.StreamOperator;
import com.alibaba.alink.operator.stream.dataproc.ToTensorStreamOp;
import com.alibaba.alink.operator.stream.source.MemSourceStreamOp;
import org.junit.Test;
public class ToTensorTest {
    @Test
    public void testToTensorBatchOp() throws Exception {
        Row[] rows = new Row[] {
            Row.of("FLOAT#6#0.0 0.1 1.0 1.1 2.0 2.1 ")
        };
        MemSourceStreamOp memSourceStreamOp = new MemSourceStreamOp(rows, new String[] {"vec"});
        memSourceStreamOp.link(
            new ToTensorStreamOp()
                .setSelectedCol("vec")
                .setTensorShape(2, 3)
                .setTensorDataType("float")
        ).print();
        StreamOperator.execute();
    }
}

运行结果

| vec | | —- |

| FLOAT#2,3#0.0 0.1 1.0 1.1 2.0 2.1 |