pytorch 之 torch.eye()函数
这个函数主要是为了生成对角线全1,其余部分全0的二维数组函数原型:result = torch.eye(n,m=None,out=None)参数解释:n:行数m:列数out:输出类型例:c = torch.eye(3)print(c)print(type(c))输出tensor([[1., 0., 0.],[0., 1., 0.],[0., 0., 1.]])<class 'torch.Te
文章共126字 · 阅读需要大约1分钟
一键AI生成摘要,助你高效阅读
问答
·
这个函数主要是为了生成对角线全1,其余部分全0的二维数组
函数原型:
result = torch.eye(n,m=None,out=None)
参数解释:
n:行数
m:列数
out:输出类型
例:
c = torch.eye(3)
print(c)
print(type(c))
输出
tensor([[1., 0., 0.],
[0., 1., 0.],
[0., 0., 1.]])
<class 'torch.Tensor'>
更多推荐
已为社区贡献7条内容
所有评论(0)