栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

动手深度学习中的python笔记(1)

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

动手深度学习中的python笔记(1)

记录一下9.7节中红线部分(不熟悉python里边一些用法)

是什么意思?

def sequence_mask(X,valid_len,value=0):
    maxlen = X.size(1)   #3
    print(maxlen)
    mask = torch.arange((maxlen),dtype=torch.float32,device=X.device)[None,:] < valid_len[:,None]
    X[~mask] = value
    return X
X = torch.tensor([[1,2,3],[4,5,6]])
sequence_mask(X,torch.tensor([1,2]))

-------------------------------------------------------------------------------------------

例1:

例2:

 综上(个人理解):

  [None,:]   可以理解为,保持最内层列数不变,将最外层元素作为基本单元进行分割。

例如(2,3) -> (1,2,3)

(3,) -> (1,3)

  [:,None] 可以理解为,保持最外层的行数不变,将最内层作为基本单元进行分割。

例如(2,3) -> (2,1,3)

(3,) ->(3,1)

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/769115.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号