← 返回列表
用于生成二进制掩码张量的方法、计算设备和存储介质
摘要文本
本发明的实施例涉及一种用于生成二进制掩码张量的方法、计算设备和存储介质。该方法包括:接收指示待生成的二进制掩码张量的概率的第一指令;针对待生成的二进制掩码张量在第一维度中的每一行,确定随机种子;至少基于概率和随机种子,确定待生成的二进制掩码张量在第二维度上的元素的值,以便生成二进制掩码张量。根据本发明提供的用于生成二进制掩码张量的方法,能够并行地确定二进制掩码张量中各行的元素,从而提高二进制掩码张量的生成效率。
申请人信息
- 申请人:北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司
- 申请人地址:100102 北京市朝阳区望京东园四区13号楼-4至33层101内10层201室
- 发明人: 北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 用于生成二进制掩码张量的方法、计算设备和存储介质 |
| 专利类型 | 发明申请 |
| 申请号 | CN202410077622.8 |
| 申请日 | 2024/1/19 |
| 公告号 | CN117591076A |
| 公开日 | 2024/2/23 |
| IPC主分类号 | G06F7/58 |
| 权利人 | 北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司 |
| 发明人 | 请求不公布姓名; 请求不公布姓名 |
| 地址 | 北京市朝阳区望京东园四区13号楼-4至33层101内10层201室; 上海市闵行区陈行公路2388号16幢13层1302室 |
专利主权项内容
1.一种用于生成二进制掩码张量的方法,其特征在于,包括:接收指示待生成的二进制掩码张量的概率的第一指令;针对所述待生成的二进制掩码张量在第一维度中的每一行,确定随机种子;至少基于所述概率和所述随机种子,确定所述待生成的二进制掩码张量在第二维度上的元素的值,以便生成二进制掩码张量。