PyTorch实战(1):矩阵运算
常见的矩阵运算:基于pytorch和numpy实现
import torch
import numpy as np
import torch.nn as nn
X = torch.rand(4, 2, 3)
linear = nn.Linear(3, 2)
"""矩阵加法"""
# Z=W+X
W = torch.randn(4, 2, 3)
Z11 = W + X
print(Z11)...
原创
2019-04-22 23:01:34 ·
5728 阅读 ·
0 评论