最新 最热

pytorch的topk()函数

pytorch.topk()用于返回Tensor中的前k个元素以及元素对应的索引值。例:

2022-09-02
1

Pytorch之permute函数

异:permute函数可以对任意高维矩阵进行转置,但没有torch.permute()这个调用方式

2022-09-02
4

PyTorch里面的torch.nn.Parameter()详解

在看过很多博客的时候发现了一个用法self.v = torch.nn.Parameter(torch.FloatTensor(hidden_size)),首先可以把这个函数理解为类型转换函数,将一个不可训练的类型Tensor转换成可以训练的类型parameter并将这个paramet...

2022-09-02
1

torch.range()和torch.arange()的区别

>>> y=torch.range(1,6)>>> ytensor([1., 2., 3., 4., 5., 6.])>>> y.dtypetorch.float32>>> z=torch.arange(1,6)>>> ztensor([1, 2, 3, 4, 5])>>> z.dtypetorch.int64总结:t......

2022-09-02
1

torch.as_tensor()

torch.as_tensor(data, dtype=None,device=None)->Tensor : 为data生成tensor。

2022-09-02
1

Pytorch之认识Variable

Tensor是Pytorch的一个完美组件,但是要构建神经网络还是远远不够的,我们需要能够计算图的Tensor,那就是Variable。Variable是对Tensor的一个封装,操作和Tensor是一样的,但是每个Variable都有三个属性,Varibale的Tensor本身...

2022-09-02
2

tf.newaxis

a的shape为(10,) b的shape为(10,1) 若b = a[tf.newaxis,:],则b的shape为(1,10)

2022-09-02
1

Expected object of scalar type Float but got scalar type Double for argument

在pytorch中float32为float类型,而float64则为double类型,注意tensor的数据类型。

2022-09-02
1

The shape of the mask [32, 8732] at index 0

如果出现类似于以上的索引错误,需要将layers/modules/multibox_loss.py 的代码按照下图修改即可。注释掉的是源代码,下面的是新加的,即交换两行代码的顺序即可。...

2022-09-02
1

TypeError: ‘Tensor‘ object does not support item assignment in TensorFlow

TypeError: 'Tensor' object does not support item assignment in TensorFlow

2022-09-02
1