Commit 07bfd45

committed

update image flattening to use reshape

1 parent 3a018eb commit 07bfd45Copy full SHA for 07bfd45

File tree

-3

lines changed

-3

lines changed

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -620,7 +620,7 @@`
`620`	`620`	`" # create reconstructed pixels\n",`
`621`	`621`	`" x = x * y[:, :, None]\n",`
`622`	`622`	`" # flatten image into a vector shape (batch_size, vector_dim)\n",`
`623`		`- " flattened_x = x.view(x.size(0), -1)\n",`
	`623`	`+ " flattened_x = x.reshape(x.size(0), -1)\n",`
`624`	`624`	`" # create reconstructed image vectors\n",`
`625`	`625`	`" reconstructions = self.linear_layers(flattened_x)\n",`
`626`	`626`	`" \n",`
`@@ -1496,7 +1496,7 @@`
`1496`	`1496`	`],`
`1497`	`1497`	`"metadata": {`
`1498`	`1498`	`"kernelspec": {`
`1499`		`- "display_name": "Python [default]",`
	`1499`	`+ "display_name": "Python 3",`
`1500`	`1500`	`"language": "python",`
`1501`	`1501`	`"name": "python3"`
`1502`	`1502`	`},`
`@@ -1510,7 +1510,7 @@`
`1510`	`1510`	`"name": "python",`
`1511`	`1511`	`"nbconvert_exporter": "python",`
`1512`	`1512`	`"pygments_lexer": "ipython3",`
`1513`		`- "version": "3.6.4"`
	`1513`	`+ "version": "3.6.10"`
`1514`	`1514`	`}`
`1515`	`1515`	`},`
`1516`	`1516`	`"nbformat": 4,`

Comments

(0)