Pytorch
masterleoo
这个作者很懒,什么都没留下…
展开
-
【Pytorch】简要介绍与总结
【Pytorch】简要介绍与总结原创 2024-06-13 14:45:06 · 377 阅读 · 0 评论 -
torch.Tensor.new_tensor(data):用于创建与data具有相同内容且与torch.Tensor有相同的dtype和device的tensor
torch.Tensor.new_tensor(data):用于创建与data具有相同内容且与torch.Tensor有相同的dtype和device的tensor原创 2024-01-12 15:16:06 · 690 阅读 · 0 评论 -
对tensor的处理函数:expand_as(尺寸扩展),nonzero(获取非零元素索引)
对tensor的处理函数:expand_as(尺寸扩展),nonzero(获取非零元素索引)原创 2023-11-23 16:56:42 · 290 阅读 · 0 评论 -
tensor和ndarray的相互转换,同时需要注意cuda和cpu的迁移
【代码】tensor和ndarray的相互转换,同时需要注意cuda和cpu的迁移。原创 2023-11-08 16:55:51 · 541 阅读 · 0 评论 -
使用new_zeros和zeros_like分别返回与输入相同尺寸/类型/device的tensor和ndarray
3)device:返回tensor的期望device,默认与输入tensor具有相同的torch.device。2)dtype:返回tensor的期望类型,默认与输入tensor具有相同的torch.dtype。1)size: list或者tuple,定义输出tensor的尺寸。dtype:输出数组的期望类型,默认与a相同。shape:输出数组的期望尺寸,默认与a相同。原创 2023-11-06 11:09:08 · 1033 阅读 · 0 评论 -
【Pycharm/Anaconda配置环境记录】
conda create --name env_name python=3.7:创建虚拟环境。查看CUDA版本可以使用nvidia-smi或者nvcc --version等命令进行查看。conda activate env_name:激活/进入该虚拟环境。conda env list:查看虚拟环境。原创 2023-09-25 19:59:02 · 353 阅读 · 0 评论 -
【Pytorch:nn.Embedding】简介以及使用方法:用于生成固定数量的具有指定维度的嵌入向量embedding vector
【Pytorch:nn.Embedding】简介以及使用方法:用于生成固定数量的具有指定维度的嵌入向量embedding vector原创 2023-08-11 16:33:36 · 1962 阅读 · 0 评论