2023年12月5日
[PyTorch] 如何判定运算维度
摘要： 实际上无论是几维，方法都是一样。假设以 torch.softmax() 为例： # 下面运行结果所使用的代码 import torch import numpy as np z = np.arange(1, 33).reshape((2, 2, 2, 4)) z = torch.tensor(z, 阅读全文
posted @ 2023-12-05 17:32 小贼的自由 阅读(16) 评论(0) 推荐(0) 编辑